changing to vectordb environment

franciscojavierarceo · franciscojavierarceo · commit 134b908595d3 · 2024-12-23T08:27:21.000-05:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/sdk/python/tests/conftest.py b/sdk/python/tests/conftest.py
@@ -191,7 +191,6 @@ def environment(request, worker_id):
         request.param,
         worker_id=worker_id,
         fixture_request=request,
-        entity_key_serialization_version=3,
     )
 
     e.setup()
diff --git a/sdk/python/tests/integration/online_store/test_universal_online.py b/sdk/python/tests/integration/online_store/test_universal_online.py
@@ -899,9 +899,8 @@ def test_retrieve_online_documents(vectordb_environment, fake_document_data):
 
 @pytest.mark.integration
 @pytest.mark.universal_online_stores(only=["milvus"])
-def test_retrieve_online_milvus_ocuments(environment, fake_document_data):
-    print(environment.online_store)
-    fs = environment.feature_store
+def test_retrieve_online_milvus_documents(vectordb_environment, fake_document_data):
+    fs = vectordb_environment.feature_store
     df, data_source = fake_document_data
     item_embeddings_feature_view = create_item_embeddings_feature_view(data_source)
     fs.apply([item_embeddings_feature_view, item()])

Original file line number	Diff line number	Diff line change
`@@ -191,7 +191,6 @@ def environment(request, worker_id):`
`191`	`191`	`request.param,`
`192`	`192`	`worker_id=worker_id,`
`193`	`193`	`fixture_request=request,`
`194`		`- entity_key_serialization_version=3,`
`195`	`194`	`)`
`196`	`195`
`197`	`196`	`e.setup()`