Skip to content

Commit 987f0fd

Browse files
marti-jorda-rocamarti-jorda-roca
authored
fix: Bytewax materialization engine fails when loading feature_store.yaml (#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]>
1 parent 2f99a61 commit 987f0fd

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

‎sdk/python/feast/infra/materialization/contrib/bytewax/Dockerfile‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,5 +25,5 @@ COPY README.md README.md
2525
# git dir to infer the version of feast we're installing.
2626
# https://github.com/pypa/setuptools_scm#usage-from-docker
2727
# I think it also assumes that this dockerfile is being built from the root of the directory.
28-
RUN --mount=source=.git,target=.git,type=bind pip3 install --no-cache-dir '.[aws,gcp,bytewax,snowflake]'
28+
RUN --mount=source=.git,target=.git,type=bind pip3 install --no-cache-dir '.[aws,gcp,bytewax,snowflake,postgres]'
2929

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

Lines changed: 2 additions & 2 deletions
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)