Add flag to drop example from PredicitonResult
Open
core
ml
P2
python
run-inference
sub-task
- Dominant language
- Java
- Stars
- 8.7k
- Forks
- 4.7k
- Avg merge
- 1d 20h
- Merged PRs (30d)
- 196
Description
RunInference currently returns both the example/input and prediction.
```
PredictionResult:
example: torch(...)
inference: torch(...)
```
Users may want the ability to only return the inference to minimize potential memory/serialization issues later in the pipeline. They can do this with a flag, and the return value may look like
```
PredictionResult:
example: None
inference: torch(...)
```
Imported from Jira [BEAM-14362](https://issues.apache.org/jira/browse/BEAM-14362). Original Jira may contain additional context.
Reported by: yeandy.
Subtask of issue #21435
Contributor guide
Assessment
This issue has not been assessed yet.