Fix document tests (#1039)

marqo-ai · Nov 15, 2024 · bc6498d · bc6498d
1 parent ef9ebb6
commit bc6498d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,7 +1,5 @@
 import pytest
 
-from tests.marqo_test import MarqoTestCase
-
 
 def pytest_addoption(parser):
     parser.addoption("--largemodel", action="store_true", default=False)

diff --git a/tests/marqo_test.py b/tests/marqo_test.py
@@ -12,15 +12,15 @@
 from starlette.applications import Starlette
 
 from marqo import config, version, tensor_search
-from marqo.tensor_search import index_meta_cache, tensor_search
-from marqo.vespa.zookeeper_client import ZookeeperClient
 from marqo.core.index_management.index_management import IndexManagement
 from marqo.core.models.marqo_index import *
 from marqo.core.models.marqo_index_request import (StructuredMarqoIndexRequest, UnstructuredMarqoIndexRequest,
                                                    FieldRequest, MarqoIndexRequest)
 from marqo.core.monitoring.monitoring import Monitoring
+from marqo.tensor_search import tensor_search
 from marqo.tensor_search.telemetry import RequestMetricsStore
 from marqo.vespa.vespa_client import VespaClient
+from marqo.vespa.zookeeper_client import ZookeeperClient
 
 
 class TestImageUrls(str, Enum):