Update to MLflow 2.7.1

crate · Sep 18, 2023 · 80f9d6f · 80f9d6f
1 parent 69899d2
commit 80f9d6f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/CHANGES.md b/CHANGES.md
@@ -2,7 +2,7 @@
 
 
 ## in progress
-
+- Update to MLflow 2.7.1
 
 ## 2023-09-12 0.1.1
 - Documentation: Improve "Container Usage" page

diff --git a/pyproject.toml b/pyproject.toml
@@ -63,7 +63,7 @@ dynamic = [
 dependencies = [
   "crash",
   "crate[sqlalchemy]",
-  "mlflow==2.6",
+  "mlflow==2.7.1",
   "sqlparse<0.5",
   "vasuki<1,>=0.4",
 ]

diff --git a/tests/test_tracking.py b/tests/test_tracking.py
@@ -55,7 +55,7 @@
 from mlflow.utils.mlflow_tags import MLFLOW_DATASET_CONTEXT, MLFLOW_RUN_NAME
 from mlflow.utils.name_utils import _GENERATOR_PREDICATES
 from mlflow.utils.os import is_windows
-from mlflow.utils.time_utils import get_current_time_millis
+from mlflow.utils.time import get_current_time_millis
 from mlflow.utils.uri import extract_db_type_from_uri
 
 from mlflow_cratedb.patch.mlflow.db_types import CRATEDB