add pgvector int test

HaoXuAI · HaoXuAI · commit 20fb9226e9c6 · 2024-04-14T18:18:20.000-07:00
diff --git a/Makefile b/Makefile
@@ -216,6 +216,25 @@ test-python-universal-postgres-online:
 				not test_snowflake" \
  			sdk/python/tests
 
+test-python-universal-pgvector-online:
+	PYTHONPATH='.' \
+		FULL_REPO_CONFIGS_MODULE=sdk.python.feast.infra.online_stores.contrib.pgvector_repo_configuration \
+		PYTEST_PLUGINS=sdk.python.tests.integration.feature_repos.universal.online_store.postgres \
+		python -m pytest -n 8 --integration \
+ 			-k "not test_universal_cli and \
+ 				not test_go_feature_server and \
+ 				not test_feature_logging and \
+				not test_reorder_columns and \
+				not test_logged_features_validation and \
+				not test_lambda_materialization_consistency and \
+				not test_offline_write and \
+				not test_push_features_to_offline_store and \
+				not gcs_registry and \
+				not s3_registry and \
+ 				not test_universal_types and \
+				not test_snowflake" \
+ 			sdk/python/tests
+
  test-python-universal-mysql-online:
 	PYTHONPATH='.' \
 		FULL_REPO_CONFIGS_MODULE=sdk.python.feast.infra.online_stores.contrib.mysql_repo_configuration \
diff --git a/sdk/python/feast/infra/online_stores/contrib/pgvector_repo_configuration.py b/sdk/python/feast/infra/online_stores/contrib/pgvector_repo_configuration.py
@@ -0,0 +1,12 @@
+from tests.integration.feature_repos.integration_test_repo_config import (
+    IntegrationTestRepoConfig,
+)
+from tests.integration.feature_repos.universal.online_store.postgres import (
+    PGVectorOnlineStoreCreator,
+)
+
+FULL_REPO_CONFIGS = [
+    IntegrationTestRepoConfig(
+        online_store="pgvector", online_store_creator=PGVectorOnlineStoreCreator
+    ),
+]
diff --git a/sdk/python/feast/infra/online_stores/contrib/postgres_repo_configuration.py b/sdk/python/feast/infra/online_stores/contrib/postgres_repo_configuration.py
@@ -2,17 +2,11 @@
     IntegrationTestRepoConfig,
 )
 from tests.integration.feature_repos.universal.online_store.postgres import (
-    PGVectorOnlineStoreCreator,
     PostgresOnlineStoreCreator,
 )
 
 FULL_REPO_CONFIGS = [
     IntegrationTestRepoConfig(
         online_store="postgres", online_store_creator=PostgresOnlineStoreCreator
-    ),
-    IntegrationTestRepoConfig(
-        online_store="pgvector", online_store_creator=PGVectorOnlineStoreCreator
-    ),
-]
-
-AVAILABLE_ONLINE_STORES = {"pgvector": PGVectorOnlineStoreCreator}
+    )
+]
