From 19fa0e57f622cc5177d54f373fe53ea155fb6ee9 Mon Sep 17 00:00:00 2001 From: Yasyf Mohamedali Date: Fri, 3 Mar 2023 22:44:06 -0800 Subject: [PATCH] Remove extraneous comma (#2993) Prevents `TypeError: category must be a Warning subclass, not 'str'`. --- op_builder/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/op_builder/utils.py b/op_builder/utils.py index 153590428..4029703e4 100644 --- a/op_builder/utils.py +++ b/op_builder/utils.py @@ -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'