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

@ -930,12 +930,12 @@ def get_nvidia_gpu_stats() -> dict[str, str]:
"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
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,