AnandInguva commented on code in PR #23266:
URL: https://github.com/apache/beam/pull/23266#discussion_r973123512
##########
sdks/python/apache_beam/ml/inference/base_test.py:
##########
@@ -251,6 +281,19 @@ def
test_run_inference_keyed_examples_with_unkeyed_model_handler(self):
| 'RunKeyed' >> base.RunInference(model_handler))
pipeline.run()
+ def test_drop_example_prediction_result(self):
+ def assert_drop_example(prediction_result):
+ assert prediction_result.example is None
+
+ pipeline = TestPipeline()
+ examples = [1, 3, 5]
+ model_handler = FakeModelHandlerWithPredictionResult()
Review Comment:
We are passing `drop_example` via `base.RunInference`, which passes it to
the `ModelHandler`.
More explanation:
https://github.com/apache/beam/pull/23266#issuecomment-1248854707
--
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]
