update

andrey-churkin · andrey-churkin · commit dded3444852d · 2025-10-30T10:18:49.000Z
diff --git a/tests/post_training/pipelines/image_classification_timm.py b/tests/post_training/pipelines/image_classification_timm.py
@@ -53,7 +53,6 @@ def prepare_model(self) -> None:
                 onnx_path,
                 export_params=True,
                 opset_version=13,
-                dynamo=False,
                 **additional_kwargs,
             )
             self.model = onnx.load(onnx_path)
diff --git a/tests/post_training/pipelines/image_classification_torchvision.py b/tests/post_training/pipelines/image_classification_torchvision.py
@@ -100,7 +100,6 @@ def prepare_model(self) -> None:
                 onnx_path,
                 export_params=True,
                 opset_version=13,
-                dynamo=False,
                 **additional_kwargs,
             )
             self.model = onnx.load(onnx_path)
diff --git a/tests/post_training/requirements.txt b/tests/post_training/requirements.txt
@@ -22,3 +22,4 @@ accelerate==1.9.0
 transformers==4.53.0
 whowhatbench @ git+https://github.com/openvinotoolkit/openvino.genai@2025.3.0.0#subdirectory=tools/who_what_benchmark
 datasets==3.6.0
+onnxscript==0.5.4

Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,6 @@ def prepare_model(self) -> None:`
`53`	`53`	`onnx_path,`
`54`	`54`	`export_params=True,`
`55`	`55`	`opset_version=13,`
`56`		`- dynamo=False,`
`57`	`56`	`**additional_kwargs,`
`58`	`57`	`)`
`59`	`58`	`self.model = onnx.load(onnx_path)`
Original file line number	Diff line number	Diff line change
`@@ -100,7 +100,6 @@ def prepare_model(self) -> None:`
`100`	`100`	`onnx_path,`
`101`	`101`	`export_params=True,`
`102`	`102`	`opset_version=13,`
`103`		`- dynamo=False,`
`104`	`103`	`**additional_kwargs,`
`105`	`104`	`)`
`106`	`105`	`self.model = onnx.load(onnx_path)`