fix: fix flake8

omnisafe/configs/on-policy/PPOLag.yaml

@@ -241,7 +241,7 @@ ShadowHandCatchOver2UnderarmSafeJoint:
       hidden_sizes: [1024, 1024, 512]
       # learning rate
       lr: 0.0006
-      
+
 SafeMetaDrive:
   # training configurations
   train_cfgs:

omnisafe/envs/__init__.py

@@ -20,11 +20,8 @@
 from omnisafe.envs.core import CMDP, env_register, make, support_envs
 from omnisafe.envs.crabs_env import CRABSEnv
 from omnisafe.envs.custom_env import CustomEnv
+from omnisafe.envs.meta_drive_env import SafetyMetaDriveEnv
 from omnisafe.envs.mujoco_env import MujocoEnv
 from omnisafe.envs.safety_gymnasium_env import SafetyGymnasiumEnv
 from omnisafe.envs.safety_gymnasium_modelbased import SafetyGymnasiumModelBased
 from omnisafe.envs.safety_isaac_gym_env import SafetyIsaacGymEnv
-
-
-with suppress(ImportError):
-    from omnisafe.envs.meta_drive_env import SafetyMetaDriveEnv

omnisafe/envs/meta_drive_env.py

@@ -21,13 +21,19 @@
 
 import numpy as np
 import torch
-from metadrive import SafeMetaDriveEnv  # pylint: disable=import-error
 
 from omnisafe.common.logger import Logger
 from omnisafe.envs.core import CMDP, env_register
 from omnisafe.typing import DEVICE_CPU
 
 
+META_DRIVE_AVAILABLE = True
+try:
+    from metadrive import SafeMetaDriveEnv
+except ImportError:
+    META_DRIVE_AVAILABLE = False
+
+
 @env_register
 class SafetyMetaDriveEnv(CMDP):
     """SafeMetaDrive Environment.
@@ -73,8 +79,10 @@ def __init__(
         self._num_envs = num_envs
         self._device = torch.device(device)
 
-        self._env = SafeMetaDriveEnv(config=kwargs.get('meta_drive_config'))
-
+        if META_DRIVE_AVAILABLE:
+            self._env = SafeMetaDriveEnv(config=kwargs.get('meta_drive_config'))
+        else:
+            raise ImportError('Please install MetaDrive to use Safe SafeMetaDrive!')
         self._num_scenarios = self._env.config['num_scenarios']
 
         self._env.logger.setLevel(logging.FATAL)