"tests/__init__.py" did not exist on "404ecbdcc61b17c7474806f5360c4fedf10cc37d"
Unverified Commit 19fa0e57 authored by Yasyf Mohamedali's avatar Yasyf Mohamedali Committed by GitHub
Browse files

Remove extraneous comma (#2993)

Prevents `TypeError: category must be a Warning subclass, not 'str'`.
parent 3a5d93bc
......@@ -157,7 +157,7 @@ def set_cuda_arch_list(cuda_dir):
# we only need to set this when CUDA is not available for cross-compilation
if not cuda_available:
warnings.warn(
'\n[extension] PyTorch did not find available GPUs on this system.\n',
'\n[extension] PyTorch did not find available GPUs on this system.\n'
'If your intention is to cross-compile, this is not an error.\n'
'By default, Colossal-AI will cross-compile for \n'
'1. Pascal (compute capabilities 6.0, 6.1, 6.2),\n'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment