Update omnisafe/utils/command_app.py

Co-authored-by: Xuehai Pan <[email protected]>
PKU-Alignment · Mar 31, 2023 · 83fa64f · 83fa64f
1 parent ad41e9e
commit 83fa64f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/omnisafe/utils/command_app.py b/omnisafe/utils/command_app.py
@@ -252,7 +252,7 @@ def benchmark(
         avaliable_gpus = list(range(torch.cuda.device_count()))
         gpu_id = list(range(*[int(i) for i in gpu_range.split(':')]))
 
-        if not set(gpu_id).issubset(set(avaliable_gpus)):
+        if not set(gpu_id).issubset(avaliable_gpus):
             warnings.warn('The GPU ID is not available, use CPU instead.', stacklevel=1)
             gpu_id = None