This is an automated email from the ASF dual-hosted git repository.
vterentev pushed a commit to branch oss-image-classification
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/oss-image-classification by
this push:
new b497badfc03 Fix input
b497badfc03 is described below
commit b497badfc03123a8f60ad0f4bd274d14dec728fe
Author: Vitaly Terentyev <[email protected]>
AuthorDate: Fri Nov 7 12:54:27 2025 +0400
Fix input
---
.../apache_beam/examples/inference/pytorch_imagenet_rightfit.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git
a/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
b/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
index e6415bc67fb..185727883e8 100644
--- a/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
+++ b/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
@@ -307,7 +307,7 @@ def run_load_pipeline(known_args, pipeline_args):
pipeline = beam.Pipeline(options=pipeline_options)
lines = (pipeline
- | 'ReadGCSFile' >>
beam.Create(list(read_gcs_file_lines(known_args.input_file)))
+ | 'ReadGCSFile' >>
beam.Create(list(read_gcs_file_lines(known_args.input)))
| 'FilterEmpty' >> beam.Filter(lambda line: line.strip()))
if known_args.rate_limit:
lines = lines | 'RateLimit' >>
beam.ParDo(RateLimitDoFn(rate_per_sec=known_args.rate_limit))
@@ -391,7 +391,7 @@ def run(argv=None, save_main_session=True,
test_pipeline=None) -> PipelineResult
if known_args.mode == 'batch':
# Batch: read URIs from a file and process
pcoll = (pipeline
- | 'ReadURIsBatch' >>
beam.Create(list(read_gcs_file_lines(known_args.input_file)))
+ | 'ReadURIsBatch' >>
beam.Create(list(read_gcs_file_lines(known_args.input)))
| 'FilterEmptyBatch' >> beam.Filter(lambda s: s.strip()))
else:
# Streaming: subscription → decode → windowing
