update import

neuralmagic · May 3, 2024 · 4bfbee4 · 4bfbee4
1 parent 4e67e67
commit 4bfbee4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/sparseml/transformers/oneshot/test_api_inputs.py b/tests/sparseml/transformers/oneshot/test_api_inputs.py
@@ -18,7 +18,7 @@
 import pytest
 
 from parameterized import parameterized_class
-from tests.sparseml.pytorch.oneshot.dataset_processing import get_data_utils
+from tests.sparseml.transformers.oneshot.dataset_processing import get_data_utils
 from tests.testing_utils import parse_params, requires_torch