Skip to content

Commit 99924a6

Browse files
omermaimEyal-Danieli
authored andcommitted
[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

5 files changed

+1054
-215
lines changed

functions/src/onnx_utils/function.yaml

Lines changed: 37 additions & 37 deletions
Large diffs are not rendered by default.

functions/src/onnx_utils/item.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ labels:
1313
author: Iguazio
1414
maintainers: []
1515
marketplaceType: ''
16-
mlrunVersion: 1.7.2
16+
mlrunVersion: 1.10.0
1717
name: onnx_utils
1818
platformVersion: 3.5.0
1919
spec:
@@ -30,13 +30,13 @@ spec:
3030
- tqdm~=4.67.1
3131
- tensorflow~=2.19.0
3232
- tf_keras~=2.19.0
33-
- torch~=2.6.0
34-
- torchvision~=0.21.0
33+
- torch~=2.8.0
34+
- torchvision~=0.23.0
3535
- onnx~=1.17.0
3636
- onnxruntime~=1.19.2
3737
- onnxoptimizer~=0.3.13
3838
- onnxmltools~=1.13.0
3939
- tf2onnx~=1.16.1
4040
- plotly~=5.23
4141
url: ''
42-
version: 1.3.0
42+
version: 1.4.0

0 commit comments

Comments
 (0)