fix: black formatting

gverkes · gverkes · commit 6374b90a7945 · 2024-02-13T19:46:08.000Z
diff --git a/src/sagemaker/local/image.py b/src/sagemaker/local/image.py
@@ -859,11 +859,7 @@ def _create_docker_host(
         if self.instance_type == "local_gpu":
             host_config["deploy"] = {
                 "resources": {
-                    "reservations": {
-                        "devices": [
-                            {"count": "all", "capabilities": ["gpu"]}
-                        ]
-                    }
+                    "reservations": {"devices": [{"count": "all", "capabilities": ["gpu"]}]}
                 }
             }
 

Original file line number	Diff line number	Diff line change
`@@ -859,11 +859,7 @@ def _create_docker_host(`
`859`	`859`	`if self.instance_type == "local_gpu":`
`860`	`860`	`host_config["deploy"] = {`
`861`	`861`	`"resources": {`
`862`		`- "reservations": {`
`863`		`- "devices": [`
`864`		`- {"count": "all", "capabilities": ["gpu"]}`
`865`		`- ]`
`866`		`- }`
	`862`	`+ "reservations": {"devices": [{"count": "all", "capabilities": ["gpu"]}]}`
`867`	`863`	`}`
`868`	`864`	`}`
`869`	`865`