update wrong import

datahub-project · Dec 20, 2024 · f94a8aa · f94a8aa
1 parent 309abf0
commit f94a8aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/mlflow.py b/metadata-ingestion/src/datahub/ingestion/source/mlflow.py
@@ -3,8 +3,8 @@
 import time
 
 from mlflow import MlflowClient
-from mlflow.entities import Run
-from mlflow.entities.model_registry import ModelVersion, RegisteredModel, Experiment
+from mlflow.entities import Run, Experiment
+from mlflow.entities.model_registry import ModelVersion, RegisteredModel
 from mlflow.store.entities import PagedList
 from pydantic.fields import Field