Add test sub class

apple · Nov 9, 2022 · 4ac3881 · 4ac3881
1 parent 6dfb6e9
commit 4ac3881
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py b/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py
@@ -5180,7 +5180,8 @@ def test_argmax(
             converter_input_type=converter_input_type,
         )
 
-class TestCopy:
+
+class TestCopy(TorchBaseTest):
     @pytest.mark.parametrize(
         "backend, rank", itertools.product(backends, list(range(1, 6))),
     )