[GitHub] [beam] ryanthompson591 commented on a diff in pull request #17527: Updates ml/inference ModelLoader to allow defining arguments to BatchElements

[GitBox]

ryanthompson591 commented on code in PR #17527:
URL: https://github.com/apache/beam/pull/17527#discussion_r863224249


##########
sdks/python/apache_beam/ml/inference/base.py:
##########
@@ -80,6 +80,10 @@ def load_model(self) -> T:
   def get_inference_runner(self) -> InferenceRunner:
     """Returns an implementation of InferenceRunner for this model."""
     raise NotImplementedError(type(self))
+    
+  def batch_elements_kwargs(self) -> Mapping[str, Any]:

Review Comment:
   You are going to need to import Mapping 



-- 
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