Skip to content
This repository has been archived by the owner on Oct 25, 2024. It is now read-only.

Commit

Permalink
Remove more ATP v2 parts
Browse files Browse the repository at this point in the history
  • Loading branch information
jaredoconnell committed Oct 12, 2023
1 parent 5e1c28f commit 4c4c708
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 14 deletions.
1 change: 0 additions & 1 deletion arcaflow_plugin_wait/arcaflow_plugin_wait.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,6 @@ class ErrorOutput:
},
)
def wait(
self,
params: InputParams,
) -> typing.Tuple[str, typing.Union[SuccessOutput, ErrorOutput]]:
"""
Expand Down
14 changes: 1 addition & 13 deletions tests/test_arcaflow_plugin_wait.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
#!/usr/bin/env python3
import unittest

from arcaflow_plugin_sdk import plugin, predefined_schemas
from arcaflow_plugin_sdk import plugin

import time
import arcaflow_plugin_wait
Expand Down Expand Up @@ -50,18 +50,6 @@ def test_functional(self):
"Waited {:0.2f} seconds after being scheduled to wait for {}"
" seconds.".format(output_data.actual_wait_seconds, WAIT_TIME)
)
# Test cancellation
input_params = arcaflow_plugin_wait.InputParams(
seconds=SKIPPED_WAIT_TIME
)
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)
run_duration = time.time() - start_time
# It starts in a cancelled state, so it should be plenty less than
# the full wait time.
self.assertLess(run_duration, SKIPPED_WAIT_TIME / 2.0)


if __name__ == '__main__':
Expand Down

0 comments on commit 4c4c708

Please sign in to comment.