Add error case

Farama-Foundation · Jan 15, 2024 · a025f65 · a025f65
1 parent e8f9bb1
commit a025f65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mo_gymnasium/envs/deep_sea_treasure/deep_sea_treasure.py b/mo_gymnasium/envs/deep_sea_treasure/deep_sea_treasure.py
@@ -137,6 +137,8 @@ def __init__(self, render_mode: Optional[str] = None, dst_map=DEFAULT_MAP, float
                 self.map_name = "convex"
             elif np.all(dst_map == CONCAVE_MAP):
                 self.map_name = "concave"
+            else:
+                raise ValueError("Invalid map")
         elif np.all(dst_map == MIRRORED_MAP):
             self.map_name = "mirrored"
         else: