State of feast

Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
feast-dev · adchia · Mar 4, 2022 · Feb 25, 2022 · Feb 26, 2022 · Feb 26, 2022
commit 2366e85d0b3a507c0ff6f53df7c266d417d7af4b
diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py
@@ -609,7 +609,7 @@ def _get_bigquery_client(project: Optional[str] = None, location: Optional[str]
 
 {{ featureview.name }}__base AS (
     SELECT
-        subquery.*,
+    subquery.*,
         entity_dataframe.entity_timestamp,
         entity_dataframe.{{featureview.name}}__entity_row_unique_id
     FROM {{ featureview.name }}__subquery AS subquery

diff --git a/sdk/python/feast/infra/passthrough_provider.py b/sdk/python/feast/infra/passthrough_provider.py
@@ -131,6 +131,8 @@ def materialize_single_feature_view(
 
         entities = []
         for entity_name in feature_view.entities:
+            print("Entity Names!!")
+            print(entity_name)
             entities.append(registry.get_entity(entity_name, project))
 
         (

@@ -1,3 +1,4 @@
+from curses import raw
 import os
 from pathlib import Path
 from typing import Any
@@ -345,9 +346,11 @@ def get_feature_server_config_from_type(feature_server_type: str):
 
 def load_repo_config(repo_path: Path) -> RepoConfig:
     config_path = repo_path / "feature_store.yaml"
-
+    print("INNNNNN")
+    print(repo_path)
     with open(config_path) as f:
         raw_config = yaml.safe_load(os.path.expandvars(f.read()))
+        print(raw_config)
         try:
             c = RepoConfig(**raw_config)
             c.repo_path = repo_path

diff --git a/sdk/python/tests/integration/feature_repos/repo_configuration.py b/sdk/python/tests/integration/feature_repos/repo_configuration.py
@@ -52,7 +52,7 @@
     "type": "redis",
     "redis_type": "redis_cluster",
     # Redis Cluster Port Forwarding is setup in "pr_integration_tests.yaml" under "Setup Redis Cluster".
-    "connection_string": "127.0.0.1:6001,127.0.0.1:6002,127.0.0.1:6003",
+    "connection_string": "127.0.0.1:7001,127.0.0.1:7002,127.0.0.1:7003",
 }
 
 # FULL_REPO_CONFIGS contains the repo configurations (e.g. provider, offline store,

diff --git a/sdk/python/tests/integration/online_store/test_universal_online.py b/sdk/python/tests/integration/online_store/test_universal_online.py
@@ -564,8 +564,8 @@ def test_online_retrieval(environment, universal_data_sources, full_feature_name
     )
 
 
-@pytest.mark.integration
-@pytest.mark.universal
+# @pytest.mark.integration
+# @pytest.mark.universal
 def test_online_store_cleanup(environment, universal_data_sources):
     """
     Some online store implementations (like Redis) keep features from different features views

diff --git a/sdk/python/tests/integration/registration/test_universal_types.py b/sdk/python/tests/integration/registration/test_universal_types.py
@@ -155,7 +155,7 @@ def test_entity_inference_types_match(offline_types_test_fixtures):
         )
 
 
-@pytest.mark.integration
+#@pytest.mark.integration
 @pytest.mark.universal
 def test_feature_get_historical_features_types_match(offline_types_test_fixtures):
     environment, config, data_source, fv = offline_types_test_fixtures
@@ -169,7 +169,7 @@ def test_feature_get_historical_features_types_match(offline_types_test_fixtures
     )
     entity = driver()
     fs.apply([fv, entity])
-
+    print("Applying")
     features = [f"{fv.name}:value"]
     entity_df = pd.DataFrame()
     entity_df["driver_id"] = (
@@ -181,9 +181,11 @@ def test_feature_get_historical_features_types_match(offline_types_test_fixtures
         ts - timedelta(hours=4),
         ts - timedelta(hours=2),
     ]
+    print("Retrieving historical features")
     historical_features = fs.get_historical_features(
         entity_df=entity_df, features=features,
     )
+    print("retrieved historical features")
     # Note: Pandas doesn't play well with nan values in ints. BQ will also coerce to floats if there are NaNs
     historical_features_df = historical_features.to_df()
     print(historical_features_df)