Skip to content

Commit d9008f1

Browse files
bytewax materialization loads yaml config correctly
Signed-off-by: marti-jorda-roca <mjorda98@gmail.com>
1 parent ea8ad17 commit d9008f1

File tree

1 file changed

+2
-2
lines changed
  • sdk/python/feast/infra/materialization/contrib/bytewax

1 file changed

+2
-2
lines changed

sdk/python/feast/infra/materialization/contrib/bytewax/dataflow.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,10 @@
1212
logging.basicConfig(level=logging.INFO)
1313

1414
with open("/var/feast/feature_store.yaml") as f:
15-
feast_config = yaml.safe_load(f)
15+
feast_config = yaml.load(f, Loader=yaml.Loader)
1616

1717
with open("/var/feast/bytewax_materialization_config.yaml") as b:
18-
bytewax_config = yaml.safe_load(b)
18+
bytewax_config = yaml.load(b, Loader=yaml.Loader)
1919

2020
config = RepoConfig(**feast_config)
2121
store = FeatureStore(config=config)

0 commit comments

Comments
 (0)