bytewax materialization loads yaml config correctly

marti-jorda-roca · marti-jorda-roca · commit 789f42a24320 · 2024-01-29T10:09:27.000+01:00
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)
