diff --git a/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py b/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py index 3576c5c9fa..7817fc0ab7 100644 --- a/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py +++ b/tests/py/dynamo/conversion/test_local_scalar_dense_aten.py @@ -1,9 +1,10 @@ import torch import torch.nn as nn -from harness import DispatchTestCase from parameterized import parameterized from torch.testing._internal.common_utils import run_tests +from .harness import DispatchTestCase + class TestLocalScalarDenseConverter(DispatchTestCase): @parameterized.expand(