Updating code to support the bytewax versions incompatibility after the upgrade. These classes ManualInputConfig and ManualOutputConfig have been removed from the new versions. hope this fixes the lint errors.

lokeshrangineni · lokeshrangineni · commit 3c89ce3e9547 · 2024-03-06T00:05:53.000-05:00
Signed-off-by: Lokesh Rangineni &lt;lokeshforjava@gmail.com&gt;

Signed-off-by: Lokesh Rangineni &lt;lokeshforjava@gmail.com&gt;
diff --git a/sdk/python/feast/infra/materialization/contrib/bytewax/bytewax_materialization_dataflow.py b/sdk/python/feast/infra/materialization/contrib/bytewax/bytewax_materialization_dataflow.py
@@ -6,8 +6,8 @@
 import pyarrow.parquet as pq
 from bytewax.dataflow import Dataflow  # type: ignore
 from bytewax.execution import cluster_main
-from bytewax.inputs import ManualInputConfig
-from bytewax.outputs import ManualOutputConfig
+from bytewax.inputs import DynamicSource
+from bytewax.outputs import DynamicSink
 
 from feast import FeatureStore, FeatureView, RepoConfig
 from feast.utils import _convert_arrow_to_proto, _run_pyarrow_field_mapping
@@ -77,7 +77,7 @@ def output_fn(mini_batch):
 
     def _run_dataflow(self):
         flow = Dataflow()
-        flow.input("inp", ManualInputConfig(self.input_builder))
+        flow.input("inp", DynamicSource(self.input_builder))
         flow.flat_map(self.process_path)
-        flow.capture(ManualOutputConfig(self.output_builder))
+        flow.output(DynamicSink(self.output_builder))
         cluster_main(flow, [], 0)
