Remove trailing spaces

This commit is contained in:
Nicolas Mowen 2023-02-03 06:50:22 -07:00 committed by GitHub
parent e71e768e4f
commit eab19af241
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -927,15 +927,15 @@ def get_nvidia_gpu_stats() -> dict[str, str]:
] ]
if ( if (
"CUDA_VISIBLE_DEVICES" in os.environ "CUDA_VISIBLE_DEVICES" in os.environ
and os.environ["CUDA_VISIBLE_DEVICES"].isdigit() and os.environ["CUDA_VISIBLE_DEVICES"].isdigit()
): ):
nvidia_smi_command.extend(["--id", os.environ['CUDA_VISIBLE_DEVICES']]) nvidia_smi_command.extend(["--id", os.environ["CUDA_VISIBLE_DEVICES"]])
elif ( elif (
"NVIDIA_VISIBLE_DEVICES" in os.environ "NVIDIA_VISIBLE_DEVICES" in os.environ
and os.environ["NVIDIA_VISIBLE_DEVICES"].isdigit() and os.environ["NVIDIA_VISIBLE_DEVICES"].isdigit()
): ):
nvidia_smi_command.extend(["--id", os.environ['NVIDIA_VISIBLE_DEVICES']]) nvidia_smi_command.extend(["--id", os.environ["NVIDIA_VISIBLE_DEVICES"]])
p = sp.run( p = sp.run(
nvidia_smi_command, nvidia_smi_command,