Atp v3 (#35)

* Update to beta ATP v3 * Update tests for new ATP run ID * Update tests for new ATP run ID * Upgrade to latest SDK * Update lock file
arcalot · Oct 31, 2023 · afdc232 · afdc232
1 parent 20bbbee
commit afdc232
Show file tree

Hide file tree

Showing 3 changed files with 51 additions and 49 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/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"

diff --git a/tests/test_arcaflow_plugin_wait.py b/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.