[onnx] have all classes that call setUp inherit from common_utils.setUp (pytorch#79543)

The remaining one is `class _TestONNXRuntime:`, which intentionally doesn't want to inherit from anything, so I left it alone.
Pull Request resolved: pytorch#79543
Approved by: https://github.com/malfet

Author: janeyx99

test/onnx/test_pytorch_onnx_caffe2.py

@@ -135,6 +135,8 @@ class TestCaffe2Backend_opset9(TestCase):
     embed_params = False
 
     def setUp(self):
+        # the following should ideally be super().setUp(), https://github.com/pytorch/pytorch/issues/79630
+        TestCase.setUp(self)
         torch.manual_seed(0)
         if torch.cuda.is_available():
             torch.cuda.manual_seed_all(0)

test/onnx/test_pytorch_onnx_onnxruntime.py

@@ -159,11 +159,10 @@ class _TestONNXRuntime:
     keep_initializers_as_inputs = True  # For IR version 3 type export.
 
     def setUp(self):
-        torch.manual_seed(0)
+        set_rng_seed(0)
         onnxruntime.set_seed(0)
         if torch.cuda.is_available():
             torch.cuda.manual_seed_all(0)
-        np.random.seed(seed=0)
         os.environ["ALLOW_RELEASED_ONNX_OPSET_ONLY"] = "0"
         self.is_script_test_enabled = True
 

test/onnx/test_verification.py

@@ -1,13 +1,13 @@
 # Owner(s): ["module: onnx"]
 
-import unittest
-
 import torch
 from torch.onnx import _experimental, verification
+from torch.testing._internal import common_utils
 
 
-class TestVerification(unittest.TestCase):
+class TestVerification(common_utils.TestCase):
     def setUp(self) -> None:
+        super().setUp()
         torch.manual_seed(0)
         if torch.cuda.is_available():
             torch.cuda.manual_seed_all(0)