Merge branch 'aqua_service_tag' of github.com:oracle/accelerated-data-science into aqua_service_tag

Author: VipulMascarenhas

tests/unitary/with_extras/model/test_model_framework_pytorch_model.py

@@ -1,14 +1,16 @@
 #!/usr/bin/env python
 
-# Copyright (c) 2021, 2023 Oracle and/or its affiliates.
+# Copyright (c) 2021, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 """Unit tests for model frameworks. Includes tests for:
- - PyTorchModel
+- PyTorchModel
 """
+
 import base64
 import os
 import shutil
+import uuid
 from io import BytesIO
 
 import numpy as np
@@ -19,7 +21,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.optim as optim
-import uuid
+
 from ads.model.framework.pytorch_model import PyTorchModel
 from ads.model.serde.model_serializer import (
     PyTorchOnnxModelSaveSERDE,
@@ -146,6 +148,9 @@ def test_serialize_with_incorrect_model_file_name_onnx(self):
                 as_onnx=True, model_file_name="model.xxx"
             )
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_serialize_using_pytorch_without_modelname(self):
         """
         Test serialize_model using pytorch without model_file_name
@@ -157,6 +162,9 @@ def test_serialize_using_pytorch_without_modelname(self):
         test_pytorch_model.serialize_model(as_onnx=False)
         assert os.path.isfile(tmp_model_dir + "model.pt")
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_serialize_using_pytorch_with_modelname(self):
         """
         Test serialize_model using pytorch with correct model_file_name
@@ -169,6 +177,9 @@ def test_serialize_using_pytorch_with_modelname(self):
         test_pytorch_model.serialize_model(as_onnx=False)
         assert os.path.isfile(tmp_model_dir + "test1.pt")
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_serialize_using_onnx_without_modelname(self):
         """
         Test serialize_model using onnx without model_file_name
@@ -183,6 +194,9 @@ def test_serialize_using_onnx_without_modelname(self):
         )
         assert os.path.exists(os.path.join(tmp_model_dir, "model.onnx"))
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_serialize_using_onnx_with_modelname(self):
         """
         Test serialize_model using onnx with correct model_file_name
@@ -200,6 +214,9 @@ def test_serialize_using_onnx_with_modelname(self):
             os.path.join(tmp_model_dir, test_pytorch_model.model_file_name)
         )
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_to_onnx(self):
         """
         Test if PytorchOnnxModelSerializer.serialize generate onnx model result.
@@ -216,6 +233,9 @@ def test_to_onnx(self):
         )
         assert os.path.exists(os.path.join(tmp_model_dir, model_file_name))
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_to_onnx_reload(self):
         """
         Test if PytorchOnnxModelSerializer.serialize generate onnx model result.
@@ -235,6 +255,9 @@ def test_to_onnx_reload(self):
             is not None
         )
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_to_onnx_without_dummy_input(self):
         """
         Test if PytorchOnnxModelSerializer.serialize raise expected error
@@ -324,6 +347,9 @@ def test_prepare_default(self):
         )
         assert os.path.exists(tmp_model_dir + "model.pt")
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_prepare_onnx(self):
         test_pytorch_model = PyTorchModel(self.myPyTorchModel, tmp_model_dir)
         test_pytorch_model.prepare(
@@ -335,6 +361,9 @@ def test_prepare_onnx(self):
         )
         assert os.path.exists(tmp_model_dir + "model.onnx")
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_prepare_onnx_with_X_sample(self):
         test_pytorch_model = PyTorchModel(self.myPyTorchModel, tmp_model_dir)
         test_pytorch_model.prepare(
@@ -346,6 +375,9 @@ def test_prepare_onnx_with_X_sample(self):
         )
         assert isinstance(test_pytorch_model.verify([1, 2, 3, 4]), dict)
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_prepare_onnx_without_input(self):
         test_pytorch_model = PyTorchModel(self.myPyTorchModel, tmp_model_dir)
         with pytest.raises(ValueError):
@@ -356,6 +388,9 @@ def test_prepare_onnx_without_input(self):
                 as_onnx=True,
             )
 
+    @pytest.mark.skip(
+        reason="ODSC-79463: Fix Missing onnxscript Dependency Causing ONNX Serialization Test Failures"
+    )
     def test_verify_onnx(self):
         """
         Test if PyTorchModel.verify in onnx serialization