From 22894053a78a7476093089229bd4f7034a3a8435 Mon Sep 17 00:00:00 2001 From: Vidur Satija Date: Sat, 10 Oct 2020 14:36:55 +0530 Subject: [PATCH 1/3] Add test for copy_ op #918 --- .../mil/frontend/torch/test/test_torch_ops.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py b/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py index deee638b7..943fab2d4 100644 --- a/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py +++ b/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py @@ -5179,3 +5179,13 @@ def test_argmax( backend=backend, converter_input_type=converter_input_type, ) + +class TestCopy: + @pytest.mark.parametrize( + "backend, rank", itertools.product(backends, list(range(1, 6))), + ) + def test_copy(self, backend, rank): + input_shape = tuple(np.random.randint(low=2, high=6, size=rank)) + + model = ModuleWrapper(function=lambda x: x.copy_()) + run_compare_torch(input_shape, model, backend=backend) From 6dfb6e96dcd01e973f70b0bdad1ed9776475e397 Mon Sep 17 00:00:00 2001 From: Toby Roseman Date: Fri, 2 Sep 2022 13:31:47 -0700 Subject: [PATCH 2/3] minor fix --- .../converters/mil/frontend/torch/test/test_torch_ops.py | 2 +- 1 file changed, 1 insertion(+), 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 index 943fab2d4..e68a4720c 100644 --- a/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py +++ b/coremltools/converters/mil/frontend/torch/test/test_torch_ops.py @@ -5188,4 +5188,4 @@ def test_copy(self, backend, rank): input_shape = tuple(np.random.randint(low=2, high=6, size=rank)) model = ModuleWrapper(function=lambda x: x.copy_()) - run_compare_torch(input_shape, model, backend=backend) + self.run_compare_torch(input_shape, model, backend=backend) From 4ac38819bac1fd3f7d5a804b2f44ea2edc207d74 Mon Sep 17 00:00:00 2001 From: Toby Roseman Date: Wed, 9 Nov 2022 10:21:45 -0800 Subject: [PATCH 3/3] Add test sub class --- .../converters/mil/frontend/torch/test/test_torch_ops.py | 3 ++- 1 file changed, 2 insertions(+), 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 index e68a4720c..523dbec51 100644 --- 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))), )