Andy Ye created BEAM-13985:
------------------------------
Summary: Implement end-to-end tests for RunInference classes
Key: BEAM-13985
URL: https://issues.apache.org/jira/browse/BEAM-13985
Project: Beam
Issue Type: Sub-task
Components: sdk-py-core
Reporter: Andy Ye
Assignee: Anand Inguva
Make calls to the RunInference classes for TFX, Pytorch, and Scikit-learn.
* For TFX, need to use their proto
Process
* Read data from GCS bucket
* Use trained model? or simple?
* Product the output predictions
* Assert if output predictions match actual
Add task for using GPU container images
--
This message was sent by Atlassian Jira
(v8.20.1#820001)