black-format

Payton Staub · Payton Staub · commit 277fc4ed9f01 · 2021-05-05T09:40:30.000-07:00
diff --git a/tests/unit/sagemaker/workflow/test_step_collections.py b/tests/unit/sagemaker/workflow/test_step_collections.py
@@ -120,6 +120,7 @@ def estimator(sagemaker_session):
         sagemaker_session=sagemaker_session,
     )
 
+
 @pytest.fixture
 def estimator_tf(sagemaker_session):
     return TensorFlow(
@@ -132,6 +133,7 @@ def estimator_tf(sagemaker_session):
         sagemaker_session=sagemaker_session,
     )
 
+
 @pytest.fixture
 def model_metrics():
     return ModelMetrics(
@@ -213,6 +215,7 @@ def test_register_model(estimator, model_metrics):
         ]
     )
 
+
 def test_register_model_tf(estimator_tf, model_metrics):
     model_data = f"s3://{BUCKET}/model.tar.gz"
     register_model = RegisterModel(

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,7 @@ def estimator(sagemaker_session):`
`120`	`120`	`sagemaker_session=sagemaker_session,`
`121`	`121`	`)`
`122`	`122`
	`123`	`+`
`123`	`124`	`@pytest.fixture`
`124`	`125`	`def estimator_tf(sagemaker_session):`
`125`	`126`	`return TensorFlow(`
`@@ -132,6 +133,7 @@ def estimator_tf(sagemaker_session):`
`132`	`133`	`sagemaker_session=sagemaker_session,`
`133`	`134`	`)`
`134`	`135`
	`136`	`+`
`135`	`137`	`@pytest.fixture`
`136`	`138`	`def model_metrics():`
`137`	`139`	`return ModelMetrics(`
`@@ -213,6 +215,7 @@ def test_register_model(estimator, model_metrics):`
`213`	`215`	`]`
`214`	`216`	`)`
`215`	`217`
	`218`	`+`
`216`	`219`	`def test_register_model_tf(estimator_tf, model_metrics):`
`217`	`220`	`model_data = f"s3://{BUCKET}/model.tar.gz"`
`218`	`221`	`register_model = RegisterModel(`