jrmccluskey commented on code in PR #34379:
URL: https://github.com/apache/beam/pull/34379#discussion_r2027178232
##########
sdks/python/apache_beam/ml/inference/base.py:
##########
@@ -339,6 +346,139 @@ def should_garbage_collect_on_timeout(self) -> bool:
return self.share_model_across_processes()
+class RemoteModelHandler(ABC, ModelHandler[ExampleT, PredictionT, ModelT]):
+ """Has the ability to call a model at a remote endpoint."""
+ def __init__(
Review Comment:
I know this is largely coming from a perspective of how to surface this for
yaml, but is this configuration consideration actually blocking here? We have
to surface all of these parameters at some point
--
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: github-unsubscr...@beam.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
