Skip to content

Commit 075a1c2

Browse files
committed
fix lint
1 parent ff90c66 commit 075a1c2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torchx/schedulers/docker_scheduler.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -299,9 +299,9 @@ def _submit_dryrun(self, app: AppDef, cfg: DockerOpts) -> AppDryRunInfo[DockerJo
299299
if resource.memMB >= 0:
300300
# To support PyTorch dataloaders we need to set /dev/shm to
301301
# larger than the 64M default.
302-
c.kwargs["mem_limit"] = c.kwargs["shm_size"] = (
303-
f"{int(resource.memMB)}m"
304-
)
302+
c.kwargs["mem_limit"] = c.kwargs[
303+
"shm_size"
304+
] = (f"{int(resource.memMB)}m")
305305
if resource.cpu >= 0:
306306
c.kwargs["nano_cpus"] = int(resource.cpu * 1e9)
307307
if resource.gpu > 0:

0 commit comments

Comments
 (0)