diff --git a/omnisafe/utils/command_app.py b/omnisafe/utils/command_app.py index 736cec16a..4de3650f9 100644 --- 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