Update test_tools.py

vertica · Nov 8, 2023 · 769507c · 769507c
1 parent f7173c7
commit 769507c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/verticapy/tests/vModel/test_tools.py b/verticapy/tests/vModel/test_tools.py
@@ -59,7 +59,7 @@ def test__is_already_stored(self, titanic_vd):
         model.fit(titanic_vd, ["age", "fare"], "survived")
         assert model._is_already_stored()
         assert (
-            model._is_already_stored(return_model_type=True).lower()
+            model._is_already_stored(return_model_type=True)[1].lower()
             == "linear_regression"
         )
         model.drop()