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

Feature/len (#68)

Browse Source 
* Add __len__ to discrete

* Add __len__ to tuple
This commit is contained in:
moto
2016-05-09 17:59:55 -07:00
committed by Greg Brockman
parent 9529fa731b
commit 705d024993
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gym/spaces/discrete.py
View File

@@ -21,3 +21,5 @@ class Discrete(Space):
return "Discrete(%d)" % self.n return "Discrete(%d)" % self.n
def __eq__(self, other): def __eq__(self, other):
return self.n == other.n return self.n == other.n
def __len__(self):
return n

3
gym/spaces/tuple_space.py
View File

@@ -7,6 +7,9 @@ class Tuple(Space):
def __init__(self, spaces): def __init__(self, spaces):
self.spaces = spaces self.spaces = spaces
def __len__(self):
return len(self.spaces)
def sample(self): def sample(self):
return tuple([space.sample() for space in self.spaces]) return tuple([space.sample() for space in self.spaces])