fix: Bytewax materialization engine fails when loading feature_store.…

…yaml (feast-dev#3912) * bytewax materialization loads yaml config correctly Signed-off-by: marti-jorda-roca <[email protected]> * added postgres dependency for SQL registries Signed-off-by: marti-jorda-roca <[email protected]> --------- Signed-off-by: marti-jorda-roca <[email protected]> Signed-off-by: Attila Toth <[email protected]>
zseta · Feb 7, 2024 · 7c3bd2e · 7c3bd2e
1 parent 8e851d9
commit 7c3bd2e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sdk/python/feast/infra/materialization/contrib/bytewax/Dockerfile b/sdk/python/feast/infra/materialization/contrib/bytewax/Dockerfile
@@ -25,5 +25,5 @@ COPY README.md README.md
 # git dir to infer the version of feast we're installing.
 # https://github.com/pypa/setuptools_scm#usage-from-docker
 # I think it also assumes that this dockerfile is being built from the root of the directory.
-RUN --mount=source=.git,target=.git,type=bind pip3 install --no-cache-dir '.[aws,gcp,bytewax,snowflake]'
+RUN --mount=source=.git,target=.git,type=bind pip3 install --no-cache-dir '.[aws,gcp,bytewax,snowflake,postgres]'
 
diff --git a/sdk/python/feast/infra/materialization/contrib/bytewax/dataflow.py b/sdk/python/feast/infra/materialization/contrib/bytewax/dataflow.py
@@ -12,10 +12,10 @@
     logging.basicConfig(level=logging.INFO)
 
     with open("/var/feast/feature_store.yaml") as f:
-        feast_config = yaml.safe_load(f)
+        feast_config = yaml.load(f, Loader=yaml.Loader)
 
         with open("/var/feast/bytewax_materialization_config.yaml") as b:
-            bytewax_config = yaml.safe_load(b)
+            bytewax_config = yaml.load(b, Loader=yaml.Loader)
 
             config = RepoConfig(**feast_config)
             store = FeatureStore(config=config)