diff --git a/examples/defining_cmps.py b/examples/defining_cmps.py index 7044267..b8e5b0b 100644 --- a/examples/defining_cmps.py +++ b/examples/defining_cmps.py @@ -9,6 +9,7 @@ from limbus.core import Component, InputParams, OutputParams, ComponentState, OutputParam, InputParam # noqa: E402 from limbus.core import Pipeline, VerboseMode # noqa: E402 + # define the components # --------------------- class Add(Component): diff --git a/tests/core/test_pipeline.py b/tests/core/test_pipeline.py index 51fe2dc..3170c5a 100644 --- a/tests/core/test_pipeline.py +++ b/tests/core/test_pipeline.py @@ -56,9 +56,9 @@ def test_pipeline_iterable(self): c2 = Constant("c2", 0) unbind = Unbind("unbind") show0 = Printer("print0") - c1.outputs.out.connect(unbind.inputs.input) + c1.outputs.out.connect(unbind.inputs.tensor) c2.outputs.out.connect(unbind.inputs.dim) - unbind.outputs.out.select(0).connect(show0.inputs.inp) + unbind.outputs.tensors.select(0).connect(show0.inputs.inp) pipeline = Pipeline() pipeline.add_nodes([c1, c2, unbind, show0]) out = pipeline.run(1)