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 (
"CUDA_VISIBLE_DEVICES" in os.environ
"CUDA_VISIBLE_DEVICES" in os.environ
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 (
"NVIDIA_VISIBLE_DEVICES" in os.environ
"NVIDIA_VISIBLE_DEVICES" in os.environ
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(
nvidia_smi_command,