diff --git a/diagrams/onprem/mlops.py b/diagrams/onprem/mlops.py index 7678d9a..dd9fd86 100644 --- a/diagrams/onprem/mlops.py +++ b/diagrams/onprem/mlops.py @@ -8,6 +8,10 @@ class _Mlops(_OnPrem): _icon_dir = "resources/onprem/mlops" +class Mlflow(_Mlops): + _icon = "mlflow.png" + + class Polyaxon(_Mlops): _icon = "polyaxon.png" diff --git a/docs/nodes/onprem.md b/docs/nodes/onprem.md index 60ad8ac..81e617a 100644 --- a/docs/nodes/onprem.md +++ b/docs/nodes/onprem.md @@ -326,6 +326,9 @@ Node classes list of onprem provider. ## onprem.mlops +Mlflow +**diagrams.onprem.mlops.Mlflow** + Polyaxon **diagrams.onprem.mlops.Polyaxon** diff --git a/resources/onprem/mlops/mlflow.png b/resources/onprem/mlops/mlflow.png new file mode 100644 index 0000000..3af24ab Binary files /dev/null and b/resources/onprem/mlops/mlflow.png differ