Commit 99924a6
[onnx_utils] updated pytorch due to security vulnerability (mlrun#968)
* Update onnx_utils dependencies and improve test robustness
- Upgrade PyTorch (2.6.0 → 2.8.0) and TorchVision (0.21.0 → 0.23.0) for
better compatibility and performance
- Update MLRun version requirement to 1.10.0 in item.yaml
- Bump function version to 1.4.0
Test improvements:
- Add environment variable validation (MLRUN_DBPATH, MLRUN_ARTIFACT_PATH)
- Add conditional test skipping based on tf2onnx availability
- Fix cleanup function to properly remove test artifacts (model.pt,
model_modules_map.json, onnx_model.onnx, etc.)
- Update deprecated artifact_path parameter to output_path
- Add explicit project context to all MLRun function calls
- Fix PyTorch test artifact path construction
* Add conftest fixture for test environment and update notebook to PyTorch demo
- Centralize test setup/cleanup in conftest autouse fixture
- Rewrite notebook demo from Keras to a working PyTorch MobileNetV2 example
* deleted iguazio credentials
* Remove conftest.py and inline fixtures into test_onnx_utils.py
Move onnx_project and test_environment fixtures directly into the test
file to reduce unnecessary indirection for a single test module.1 parent 04bf508 commit 99924a6
File tree
5 files changed
+1054
-215
lines changed- functions/src/onnx_utils
5 files changed
+1054
-215
lines changedLarge diffs are not rendered by default.
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
13 | 13 | | |
14 | 14 | | |
15 | 15 | | |
16 | | - | |
| 16 | + | |
17 | 17 | | |
18 | 18 | | |
19 | 19 | | |
| |||
30 | 30 | | |
31 | 31 | | |
32 | 32 | | |
33 | | - | |
34 | | - | |
| 33 | + | |
| 34 | + | |
35 | 35 | | |
36 | 36 | | |
37 | 37 | | |
38 | 38 | | |
39 | 39 | | |
40 | 40 | | |
41 | 41 | | |
42 | | - | |
| 42 | + | |
0 commit comments