You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@beam.apache.org by tv...@apache.org on 2022/06/07 20:48:16 UTC

[beam] 01/01: [BEAM-14337] Support batched key examples and non-batchable kwargs params for RunInference models #21733

This is an automated email from the ASF dual-hosted git repository.

tvalentyn pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git

commit 36cdcc6b9da5357b5a0c2b6b9cda11438edfd52e
Merge: e95ef975a3c f647dbdf8c9
Author: tvalentyn <tv...@users.noreply.github.com>
AuthorDate: Tue Jun 7 22:48:09 2022 +0200

     [BEAM-14337] Support batched key examples and non-batchable kwargs params for RunInference models #21733

 sdks/python/apache_beam/ml/inference/base.py       |  20 +-
 sdks/python/apache_beam/ml/inference/base_test.py  |  24 ++-
 .../apache_beam/ml/inference/pytorch_inference.py  |  53 ++++-
 .../ml/inference/pytorch_inference_test.py         | 228 +++++++++++++++++----
 .../apache_beam/ml/inference/sklearn_inference.py  |   6 +-
 5 files changed, 265 insertions(+), 66 deletions(-)