AnandInguva commented on code in PR #22164:
URL: https://github.com/apache/beam/pull/22164#discussion_r929612584
##########
sdks/python/apache_beam/examples/inference/pytorch_image_classification.py:
##########
@@ -72,6 +74,7 @@ def parse_known_args(argv):
"""Parses args for the workflow."""
parser = argparse.ArgumentParser()
parser.add_argument(
+ '--input_file',
Review Comment:
Removed input_file.
##########
sdks/python/apache_beam/examples/inference/pytorch_image_classification.py:
##########
@@ -95,21 +98,28 @@ def parse_known_args(argv):
return parser.parse_known_args(argv)
-def run(argv=None, model_class=None, model_params=None,
save_main_session=True):
+def run(
+ argv=None,
+ model_class=None,
+ model_params=None,
+ save_main_session=True,
+ test_pipeline=None) -> PipelineResult:
"""
Args:
argv: Command line arguments defined for this example.
model_class: Reference to the class definition of the model.
model_params: Parameters passed to the constructor of the model_class.
These will be used to instantiate the model object in the
RunInference API.
+ test_pipeline: used for internal testing. No backwards-compatibility,
Review Comment:
Done
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
