Fix UT

open-compass · Aug 2, 2023 · 326da8c · 326da8c
1 parent ba77857
commit 326da8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opencompass/runners/local.py b/opencompass/runners/local.py
@@ -66,7 +66,7 @@ def launch(self, tasks: List[Dict[str, Any]]) -> List[Tuple[str, int]]:
             if 'CUDA_VISIBLE_DEVICES' in os.environ:
                 all_gpu_ids = [
                     int(i) for i in re.findall(
-                        '(?<!-)\d+', os.getenv('CUDA_VISIBLE_DEVICES'))
+                        r'(?<!-)\d+', os.getenv('CUDA_VISIBLE_DEVICES'))
                 ]
             else:
                 all_gpu_ids = list(range(torch.cuda.device_count()))