Fix flake8

Author: Payton Staub

tests/unit/sagemaker/workflow/test_callback_step.py

@@ -22,12 +22,12 @@
 from sagemaker.workflow.pipeline import Pipeline
 from sagemaker.workflow.callback_step import CallbackStep, CallbackOutput, CallbackOutputTypeEnum
 
-from tests.unit.sagemaker.workflow.helpers import ordered
 
 @pytest.fixture
 def sagemaker_session_mock():
     return Mock()
 
+
 def test_callback_step():
     param = ParameterInteger(name="MyInt")
     outputParam1 = CallbackOutput(output_name="output1", output_type=CallbackOutputTypeEnum.String)
@@ -52,6 +52,7 @@ def test_callback_step():
         "Arguments": {"arg1": "foo", "arg2": 5, "arg3": param},
     }
 
+
 def test_callback_step_output_expr():
     param = ParameterInteger(name="MyInt")
     outputParam1 = CallbackOutput(output_name="output1", output_type=CallbackOutputTypeEnum.String)
@@ -67,6 +68,7 @@ def test_callback_step_output_expr():
     assert cb_step.properties.Outputs['output1'].expr == {"Get": "Steps.MyCallbackStep.OutputParameters['output1']"}
     assert cb_step.properties.Outputs['output2'].expr == {"Get": "Steps.MyCallbackStep.OutputParameters['output2']"}
 
+
 def test_pipeline_interpolates_callback_outputs():
     parameter = ParameterString("MyStr")
     outputParam1 = CallbackOutput(output_name="output1", output_type=CallbackOutputTypeEnum.String)
@@ -94,48 +96,47 @@ def test_pipeline_interpolates_callback_outputs():
     )
 
     assert json.loads(pipeline.definition()) == {
-            "Version": "2020-12-01",
-            "Metadata": {},
-            "Parameters": [{"Name": "MyStr", "Type": "String"}],
-            "PipelineExperimentConfig": {
-                "ExperimentName": {"Get": "Execution.PipelineName"},
-                "TrialName": {"Get": "Execution.PipelineExecutionId"},
+        "Version": "2020-12-01",
+        "Metadata": {},
+        "Parameters": [{"Name": "MyStr", "Type": "String"}],
+        "PipelineExperimentConfig": {
+            "ExperimentName": {"Get": "Execution.PipelineName"},
+            "TrialName": {"Get": "Execution.PipelineExecutionId"},
+        },
+        "Steps": [
+            {
+                "Name": "MyCallbackStep1",
+                "Type": "Callback",
+                "Arguments": {
+                    "arg1": "foo"
+                },
+                "DependsOn": [
+                    "TestStep"
+                ],
+                "SqsQueueUrl": "https://sqs.us-east-2.amazonaws.com/123456789012/MyQueue",
+                "OutputParameters": [
+                    {
+                        "OutputName": "output1",
+                        "OutputType": "String"
+                    }
+                ]
             },
-            "Steps": [
-                {
-                    "Name": "MyCallbackStep1",
-                    "Type": "Callback",
-                    "Arguments": {
-                        "arg1": "foo"
-                    },
-                    "DependsOn": [
-                        "TestStep"
-                    ],
-                    "SqsQueueUrl": "https://sqs.us-east-2.amazonaws.com/123456789012/MyQueue",
-                    "OutputParameters": [
-                        {
-                            "OutputName": "output1",
-                            "OutputType": "String"
-                        }
-                    ]
+            {
+                "Name": "MyCallbackStep2",
+                "Type": "Callback",
+                "Arguments": {
+                    "arg1": {"Get": "Steps.MyCallbackStep1.OutputParameters['output1']"}
                 },
-                {
-                    "Name": "MyCallbackStep2",
-                    "Type": "Callback",
-                    "Arguments": {
-                        "arg1": { "Get": "Steps.MyCallbackStep1.OutputParameters['output1']"}
-                    },
-                    "DependsOn": [
-                        "TestStep"
-                    ],
-                    "SqsQueueUrl": "https://sqs.us-east-2.amazonaws.com/123456789012/MyQueue",
-                    "OutputParameters": [
-                        {
-                            "OutputName": "output2",
-                            "OutputType": "String"
-                        }
-                    ]
-                }
-            ]
-        }
-    
+                "DependsOn": [
+                    "TestStep"
+                ],
+                "SqsQueueUrl": "https://sqs.us-east-2.amazonaws.com/123456789012/MyQueue",
+                "OutputParameters": [
+                    {
+                        "OutputName": "output2",
+                        "OutputType": "String"
+                    }
+                ]
+            }
+        ]
+    }