fix: invert the condition for assertions

instadeepai · Oct 25, 2024 · b3d2495 · b3d2495
1 parent 67878f4
commit b3d2495
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jumanji/environments/routing/lbf/generator.py b/jumanji/environments/routing/lbf/generator.py
@@ -55,10 +55,10 @@ def __init__(
         assert num_food > 0, "Number of food items must be positive."
         assert max_agent_level >= 2, "Maximum agent level must be at least 2."
 
-        min_required_cells = num_agents + num_food * 3
+        min_required_cells = num_agents + num_food
         assert (
             grid_size**2
-        ) * 0.4 <= min_required_cells, r"""Ensure at least 40% of the grid cells remain unoccupied
+        ) * 0.6 >= min_required_cells, r"""Ensure at least 40% of the grid cells remain unoccupied
         to facilitate smooth placement and movement of agents and food items."""
 
         self.grid_size = grid_size