diff --git a/tests/py/dynamo/converters/test_neg_aten.py b/tests/py/dynamo/conversion/test_neg_aten.py similarity index 95% rename from tests/py/dynamo/converters/test_neg_aten.py rename to tests/py/dynamo/conversion/test_neg_aten.py index d5d805f9c2..218c951f1a 100644 --- a/tests/py/dynamo/converters/test_neg_aten.py +++ b/tests/py/dynamo/conversion/test_neg_aten.py @@ -3,7 +3,8 @@ from parameterized import parameterized from torch.testing._internal.common_utils import run_tests from torch_tensorrt import Input -from torch_tensorrt.dynamo.test_utils import DispatchTestCase + +from .harness import DispatchTestCase class TestNegConverter(DispatchTestCase):