Skip to content

Commit 6c0f7fd

Browse files
committed
fix line too long in local_gpu case
1 parent 2c37ea2 commit 6c0f7fd

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/local/image.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -759,7 +759,9 @@ def _create_docker_host(self, host, environment, optml_subdirs, command, volumes
759759
# for GPU support pass in nvidia as the runtime, this is equivalent
760760
# to setting --runtime=nvidia in the docker commandline.
761761
if self.instance_type == "local_gpu":
762-
host_config["deploy"] = {"resources": {"reservations": {"devices": [{"capabilities": ["gpu"]}]}}}
762+
host_config["deploy"] = {
763+
"resources": {"reservations": {"devices": [{"capabilities": ["gpu"]}]}}
764+
}
763765

764766
if command == "serve":
765767
serving_port = (

0 commit comments

Comments
 (0)