gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-30 01:50:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix for issue 1256 (Box(low=0, high=255, dtype='uint8').sample() returned zeros) (#1307)

Browse Source
This commit is contained in:
pzhokhov
2019-02-05 17:49:29 -08:00
committed by GitHub
parent 4ceff7dc09
commit 3067a0b890
2 changed files with 47 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gym/spaces/box.py
View File

@@ -41,7 +41,8 @@ class Box(Space):
self.np_random.seed(seed)
def sample(self):
return self.np_random.uniform(low=self.low, high=self.high + (0 if self.dtype.kind == 'f' else 1), size=self.low.shape).astype(self.dtype)
high = self.high if self.dtype.kind == 'f' else self.high.astype('int64') + 1
return self.np_random.uniform(low=self.low, high=high, size=self.low.shape).astype(self.dtype)
def contains(self, x):
return x.shape == self.shape and (x >= self.low).all() and (x <= self.high).all()