fix pipeline tests

lenskit · Jan 8, 2025 · 42a06c8 · 42a06c8
1 parent 31c0f34
commit 42a06c8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/lenskit/tests/pipeline/test_pipeline.py b/lenskit/tests/pipeline/test_pipeline.py
@@ -12,7 +12,8 @@
 
 from pytest import raises, warns
 
-from lenskit.pipeline import InputNode, Node, Pipeline, PipelineError
+from lenskit.pipeline import Pipeline, PipelineError
+from lenskit.pipeline.nodes import InputNode, Node
 from lenskit.pipeline.types import TypecheckWarning
 
 

diff --git a/lenskit/tests/pipeline/test_save_load.py b/lenskit/tests/pipeline/test_save_load.py
@@ -13,10 +13,10 @@
 
 from pytest import fail, warns
 
-from lenskit.pipeline import InputNode, Node, Pipeline, PipelineWarning
+from lenskit.pipeline import Pipeline, PipelineWarning
 from lenskit.pipeline.components import Component
 from lenskit.pipeline.config import PipelineConfig
-from lenskit.pipeline.nodes import ComponentNode
+from lenskit.pipeline.nodes import ComponentNode, InputNode
 
 _log = logging.getLogger(__name__)