[
https://issues.apache.org/jira/browse/BEAM-14044?focusedWorklogId=765202&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-765202
]
ASF GitHub Bot logged work on BEAM-14044:
-----------------------------------------
Author: ASF GitHub Bot
Created on: 02/May/22 23:21
Start Date: 02/May/22 23:21
Worklog Time Spent: 10m
Work Description: zwestrick commented on code in PR #17527:
URL: https://github.com/apache/beam/pull/17527#discussion_r863254552
##########
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:
Added
Issue Time Tracking
-------------------
Worklog Id: (was: 765202)
Time Spent: 0.5h (was: 20m)
> Hook In Batching DoFn Apis to RunInference
> ------------------------------------------
>
> Key: BEAM-14044
> URL: https://issues.apache.org/jira/browse/BEAM-14044
> Project: Beam
> Issue Type: Sub-task
> Components: sdk-py-core
> Reporter: Ryan Thompson
> Assignee: Brian Hulette
> Priority: P2
> Time Spent: 0.5h
> Remaining Estimate: 0h
>
> Hook into the batching DoFn APIs to the base RunInference interface.
> We should also investigate what defaults we should set for batching, and
> perhaps make that part of the API.
> See
> [s.apache.org/batched-dofns|https://www.google.com/url?q=http://s.apache.org/batched-dofns&sa=D&source=docs&ust=1646063987404027&usg=AOvVaw1VO9QgWlbAhx0Rh2Bzl1nw]
> for more details.
--
This message was sent by Atlassian Jira
(v8.20.7#820007)
