Fix space utils for Discrete with non-zero start (#2645)

* Fix flatten utils to handle Discrete.start

* Fix vector space utils to handle Discrete.start

* More granular dispatch in vector utils

* Fix Box including the high end of the interval
This commit is contained in:
Tristan Deleu
2022-03-04 15:17:16 -05:00
committed by GitHub
parent 108f32c743
commit e671aa168c
6 changed files with 50 additions and 30 deletions

View File

@@ -18,6 +18,7 @@ spaces = [
Box(low=0, high=255, shape=(), dtype=np.uint8),
Box(low=0, high=255, shape=(32, 32, 3), dtype=np.uint8),
Discrete(2),
Discrete(5, start=-2),
Tuple((Discrete(3), Discrete(5))),
Tuple(
(