Atp v3

pyproject.toml

@@ -18,7 +18,7 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.9"
-arcaflow-plugin-sdk = "^0.11.0"
+arcaflow-plugin-sdk = "~0.13.0"
 
 [tool.poetry.dev-dependencies]
 coverage = "^7.0.0"

tests/test_arcaflow_plugin_wait.py

@@ -42,7 +42,7 @@ def test_functional(self):
             seconds=WAIT_TIME
         )
         wait_step = arcaflow_plugin_wait.WaitStep()
-        output_id, output_data = wait_step.wait(input_params)
+        output_id, output_data = wait_step.wait(self.id(), input_params)
 
         self.assertEqual("success", output_id)
         self.assertEqual(
@@ -57,7 +57,7 @@ def test_functional(self):
         wait_step = arcaflow_plugin_wait.WaitStep()
         wait_step.cancel_step(wait_step, predefined_schemas.cancelInput())
         start_time = time.time()
-        output_id, output_data = wait_step.wait(input_params)
+        output_id, output_data = wait_step.wait(self.id(), input_params)
         run_duration = time.time() - start_time
         # It starts in a cancelled state, so it should be plenty less than
         # the full wait time.