fix types

chanzuckerberg · Aug 14, 2023 · 5a87678 · 5a87678
1 parent d5834b4
commit 5a87678
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_wdl.py b/test/test_wdl.py
@@ -325,7 +325,7 @@ def _wait_sfn(
         sfn_arn: str,
         n_stages: int = 1,
         expect_success: bool = True
-    ) -> Tuple[str, Dict, List[Dict]]:
+    ) -> Tuple[str, Dict, List[Dict], List[str]]:
         execution_name = "swipe-test-{}".format(int(time.time()))
         res = self.sfn.start_execution(
             stateMachineArn=sfn_arn, name=execution_name, input=json.dumps(sfn_input)
@@ -338,9 +338,9 @@ def _wait_sfn(
             time.sleep(10)
             description = self.sfn.describe_execution(executionArn=arn)
 
-        while messages := self.retrieve_message(self.step_change_queue_url):
+        while step_messages := self.retrieve_message(self.step_change_queue_url):
             step_notifications.append(
-                messages
+                step_messages
             )
 
         print("printing execution history", file=sys.stderr)