apache / apache/beam

[Feature Request]: Python RunInference transform should be Schema-Aware

Open
#23,467 3 comments 1 reaction 0 assignees View on GitHub
ml new feature P2 run-inference
Dominant language
Java
Stars
8.7k
Forks
4.7k
Avg merge
1d 20h
Merged PRs (30d)
196

Description

### What would you like to happen?

Currently, Python RunInference transform takes [parameterized input/output types](https://github.com/apache/beam/blob/master/sdks/python/apache_beam/ml/inference/base.py#L59) that may or may not be Schema-Aware. This makes it harder to use this transform for cases where we need a Schema-Aware/portable type (for example, Multi-language, SQL). For such cases, the transform may or may not be usable depending on the ModelHandler used. I think we need to update this transform so that users can directly operate on Schema-Aware input/output PCollections.

### Issue Priority

Priority: 2

### Issue Component

Component: run-inference

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.