From 80f9d6fb4577245e8f72fafd3ae0c3ddfcee2361 Mon Sep 17 00:00:00 2001
From: Andreas Motl <andreas.motl@crate.io>
Date: Mon, 18 Sep 2023 13:39:27 +0200
Subject: [PATCH] Update to MLflow 2.7.1

---
 CHANGES.md             | 2 +-
 pyproject.toml         | 2 +-
 tests/test_tracking.py | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/CHANGES.md b/CHANGES.md
index 4882e89..f38cf44 100644
--- 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
index 6f38052..1edb745 100644
--- 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
index 92ef0e2..508fbaa 100644
--- 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