fix tests

dlt-hub · Jan 13, 2024 · 69aae82 · 69aae82
1 parent 1a4563d
commit 69aae82
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/load/weaviate/utils.py b/tests/load/weaviate/utils.py
@@ -79,6 +79,8 @@ def delete_classes(p, class_list):
 
 def drop_active_pipeline_data() -> None:
     def schema_has_classes(client):
+        if not hasattr(client, "db_client"):
+            return None
         schema = client.db_client.schema.get()
         return schema["classes"]