Skip to content

[tmva][sofie][ci] Enable TMVA SOFIE on all Linux platforms and add Python 3.14 support for SOFIE tests#21483

Open
guitargeek wants to merge 2 commits intoroot-project:masterfrom
guitargeek:sofie_ci
Open

[tmva][sofie][ci] Enable TMVA SOFIE on all Linux platforms and add Python 3.14 support for SOFIE tests#21483
guitargeek wants to merge 2 commits intoroot-project:masterfrom
guitargeek:sofie_ci

Conversation

@guitargeek
Copy link
Contributor

We were inconsistent about where we enabled tmva-sofie, and the newer Linux platforms didn't have it. It would be good to enable it also for new Fedoras and Ubuntus to get test coverage of new developments.

@guitargeek guitargeek self-assigned this Mar 4, 2026
@guitargeek guitargeek requested a review from dpiparo as a code owner March 4, 2026 09:20
@guitargeek guitargeek requested a review from lmoneta as a code owner March 4, 2026 12:42
@guitargeek guitargeek changed the title [ci] Enable TMVA SOFIE on all Linux platforms [ci] Enable TMVA SOFIE on all Linux platforms and add Python 3.14 support for SOFIE tests Mar 4, 2026
@guitargeek guitargeek changed the title [ci] Enable TMVA SOFIE on all Linux platforms and add Python 3.14 support for SOFIE tests [tmva][sofie][ci] Enable TMVA SOFIE on all Linux platforms and add Python 3.14 support for SOFIE tests Mar 4, 2026
PyTorch’s new torch.export/Dynamo-based ONNX exporter fails under Python 3.14
(onnxscript typeinfo/typing.Union error), causing the SOFIE model generator
scripts to crash before writing the .onnx files and making the tests fail.

This commit suggests to make use of the dynamo exporter on the
interpreter version:

  * import sys and compute `dynamo_export` flag from `sys.version_info`

  * pass `dynamo=dynamo_export` to `torch.onnx.export(...)`

  * keep existing special-cases (e.g. batchnorm still forces legacy export)

This restores SOFIE ONNX generation and fixes Conv1d/Conv3d (and related) test
failures on Python 3.14 while preserving the current behavior on other Python
versions.
We were inconsistent about where we enabled `tmva-sofie`, and the newer
Linux platforms didn't have it. It would be good to enable it also for
new Fedoras and Ubuntus to get test coverage of new developments.
@github-actions
Copy link

github-actions bot commented Mar 5, 2026

Test Results

    22 files      22 suites   3d 7h 24m 48s ⏱️
 3 831 tests  3 824 ✅   1 💤 6 ❌
76 740 runs  76 606 ✅ 128 💤 6 ❌

For more details on these failures, see this check.

Results for commit 24c3c4c.

♻️ This comment has been updated with latest results.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant