pytorch-lightning

577c04d0 - Allow to return `ONNXProgram` when calling `to_onnx(dynamo=True)` (#20811)

Allow to return `ONNXProgram` when calling `to_onnx(dynamo=True)` (#20811) * feat: return `ONNXProgram` when exporting with dynamo=True. * test: add to_onnx(dynamo=True) unittests. * fix: add ignore filter in pyproject.toml * fix: change the return type annotation of `to_onnx`. * test: add parametrized `dynamo` to test `test_if_inference_output_is_valid`. * test: add difference check in `test_model_return_type`. * fix: fix unittest. * test: add test `test_model_onnx_export_missing_onnxscript`. * feat: enable ONNXProgram export on torch 2.5.0 * extensions --------- Co-authored-by: Jirka B <j.borovec+github@gmail.com> Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>