apache / apache/beam

Add flag to drop example from PredicitonResult

Open
#21,444 11 comments 0 reactions 1 assignee Claimed by @AnandInguva View on GitHub
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

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.