add StepRunInputResponse

zenml-io · Nov 6, 2024 · f9c4455 · f9c4455
1 parent 5b6bf57
commit f9c4455
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/integration/functional/artifacts/test_artifacts_linage.py b/tests/integration/functional/artifacts/test_artifacts_linage.py
@@ -319,25 +319,25 @@ def my_pipeline():
         prr: PipelineRunResponse = my_pipeline.with_options(
             enable_cache=cache
         )()
-        assert len(prr.steps["producer_step"].input_types) == 0
+        assert len(prr.steps["producer_step"].inputs) == 0
         assert (
-            prr.steps["cs1"].input_types["shared_name"]
+            prr.steps["cs1"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.STEP_OUTPUT
         )
         assert (
-            prr.steps["cs2"].input_types["shared_name"]
+            prr.steps["cs2"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.EXTERNAL
         )
         assert (
-            prr.steps["cs3"].input_types["shared_name"]
+            prr.steps["cs3"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.LAZY_LOADED
         )
         assert (
-            prr.steps["mcsl"].input_types["shared_name"]
+            prr.steps["mcsl"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.MANUAL
         )
         assert (
-            prr.steps["mcsc"].input_types["shared_name"]
+            prr.steps["mcsc"].inputs["shared_name"].input_type
             == StepRunInputArtifactType.MANUAL
         )