[GitHub] [beam] riteshghorse commented on a diff in pull request #24911: Support ONNX runtime in RunInference API

["riteshghorse (via GitHub)" <gi...@apache.org>]

riteshghorse commented on code in PR #24911:
URL: https://github.com/apache/beam/pull/24911#discussion_r1101954405


##########
sdks/python/tox.ini:
##########
@@ -329,3 +331,20 @@ commands =
   # Run all PyTorch unit tests
   # Allow exit code 5 (no tests run) so that we can run this command safely on arbitrary subdirectories.
   /bin/sh -c 'pytest -o junit_suite_name={envname} --junitxml=pytest_{envname}.xml -n 6 -m uses_pytorch {posargs}; ret=$?; [ $ret = 5 ] && exit 0 || exit $ret'
+
+[testenv:py{38}-onnx-{113}]
+deps =
+  onnxruntime==1.13.1
+  pandas==1.5.2
+  torch==1.13.1
+  tensorflow==2.11.0
+  tf2onnx==1.13.0
+  skl2onnx==1.13
+  transformers==4.25.1
+extras = test,gcp
+commands =
+  # Log onnx version for debugging
+  /bin/sh -c "pip freeze | grep -E onnx"
+  # Run all PyTorch unit tests

Review Comment:
   ```suggestion
     # Run all ONNX unit tests
   ```



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: github-unsubscribe@beam.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org