Tests: Fix import error

crate · Sep 9, 2023 · ade3468 · ade3468
1 parent 36c6def
commit ade3468
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_tracking.py b/tests/test_tracking.py
@@ -69,7 +69,7 @@
 from mlflow.utils.time_utils import get_current_time_millis
 from mlflow.utils.uri import extract_db_type_from_uri
 
-from mlflow_cratedb.adapter.db import CRATEDB
+from mlflow_cratedb.patch.mlflow.db_types import CRATEDB
 from .abstract import AbstractStoreTest
 from .util import invoke_cli_runner, assert_dataset_inputs_equal