From ade346802dbb9964fad19704825739547715b095 Mon Sep 17 00:00:00 2001
From: Andreas Motl <andreas.motl@crate.io>
Date: Sat, 9 Sep 2023 15:48:55 +0200
Subject: [PATCH] Tests: Fix import error

---
 tests/test_tracking.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_tracking.py b/tests/test_tracking.py
index 4f8484c..e3852f9 100644
--- 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