fix import

neuralmagic · Jun 27, 2023 · d09e33a · d09e33a
1 parent 8f95888
commit d09e33a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/deepsparse/pipelines/test_dynamic_import.py b/tests/deepsparse/pipelines/test_dynamic_import.py
@@ -15,7 +15,8 @@
 import os
 
 import pytest
-from deepsparse.pipeline import _REGISTERED_PIPELINES, Pipeline
+from deepsparse.base_pipeline import _REGISTERED_PIPELINES
+from deepsparse.pipeline import Pipeline
 from deepsparse.tasks import _split_dir_and_name, dynamic_import_task