Fix negative value sampling in MultiDiscrete with non-zero start (#562)

Co-authored-by: Mark Towers <[email protected]>
Farama-Foundation · Jul 3, 2023 · 6126e5f · 6126e5f
1 parent f399319
commit 6126e5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gymnasium/spaces/multi_discrete.py b/gymnasium/spaces/multi_discrete.py
@@ -143,9 +143,9 @@ def _apply_mask(
 
             return np.array(_apply_mask(mask, self.nvec, self.start), dtype=self.dtype)
 
-        return (self.np_random.random(self.nvec.shape) * self.nvec + self.start).astype(
+        return (self.np_random.random(self.nvec.shape) * self.nvec).astype(
             self.dtype
-        )
+        ) + self.start
 
     def contains(self, x: Any) -> bool:
         """Return boolean specifying if x is a valid member of this space."""