Fix another thing lol

Farama-Foundation · Jan 15, 2024 · 70fc921 · 70fc921
1 parent a025f65
commit 70fc921
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mo_gymnasium/envs/deep_sea_treasure/deep_sea_treasure.py b/mo_gymnasium/envs/deep_sea_treasure/deep_sea_treasure.py
@@ -144,7 +144,7 @@ def __init__(self, render_mode: Optional[str] = None, dst_map=DEFAULT_MAP, float
         else:
             raise ValueError("Invalid map")
         print(f"Using {self.map_name} map")
-        self._pareto_front = CONVEX_FRONT if np.all(dst_map == DEFAULT_MAP) else CONCAVE_FRONT
+        self._pareto_front = CONVEX_FRONT if self.map_name == "convex" else CONCAVE_FRONT
 
         self.dir = {
             0: np.array([-1, 0], dtype=np.int32),  # up