formatting changes

shreyapandit · shreyapandit · commit 3535d91e9bbe · 2021-09-14T16:47:16.000-07:00
diff --git a/tests/unit/sagemaker/image_uris/test_algos.py b/tests/unit/sagemaker/image_uris/test_algos.py
@@ -33,7 +33,7 @@
     "randomcutforest",
     "semantic-segmentation",
     "seq2seq",
-    "lda"
+    "lda",
 )
 ALGO_REGIONS_AND_ACCOUNTS = (
     {
diff --git a/tests/unit/sagemaker/image_uris/test_data_wrangler.py b/tests/unit/sagemaker/image_uris/test_data_wrangler.py
@@ -43,11 +43,11 @@
 
 def test_data_wrangler_ecr_uri():
     for region in DATA_WRANGLER_ACCOUNTS.keys():
-            actual_uri = image_uris.retrieve("data-wrangler", region=region)
-            expected_uri = expected_uris.algo_uri(
-                "sagemaker-data-wrangler-container",
-                DATA_WRANGLER_ACCOUNTS[region],
-                region,
-                version="1.x",
-            )
-            assert expected_uri == actual_uri
+        actual_uri = image_uris.retrieve("data-wrangler", region=region)
+        expected_uri = expected_uris.algo_uri(
+            "sagemaker-data-wrangler-container",
+            DATA_WRANGLER_ACCOUNTS[region],
+            region,
+            version="1.x",
+        )
+        assert expected_uri == actual_uri
diff --git a/tests/unit/sagemaker/image_uris/test_neo.py b/tests/unit/sagemaker/image_uris/test_neo.py
@@ -95,6 +95,7 @@ def _test_inferentia_framework_uris(framework, version):
         )
         assert expected == uri
 
+
 def test_inferentia_mxnet(inferentia_mxnet_version):
     _test_inferentia_framework_uris("mxnet", inferentia_mxnet_version)
 

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@`
`33`	`33`	`"randomcutforest",`
`34`	`34`	`"semantic-segmentation",`
`35`	`35`	`"seq2seq",`
`36`		`- "lda"`
	`36`	`+ "lda",`
`37`	`37`	`)`
`38`	`38`	`ALGO_REGIONS_AND_ACCOUNTS = (`
`39`	`39`	`{`
Original file line number	Diff line number	Diff line change
`@@ -95,6 +95,7 @@ def _test_inferentia_framework_uris(framework, version):`
`95`	`95`	`)`
`96`	`96`	`assert expected == uri`
`97`	`97`
	`98`	`+`
`98`	`99`	`def test_inferentia_mxnet(inferentia_mxnet_version):`
`99`	`100`	`_test_inferentia_framework_uris("mxnet", inferentia_mxnet_version)`
`100`	`101`