Update Composite spaces with full coverage (#3047)

* Update composite space and tests

* Pre-commit

* pyright

* Fix pyright

* retrigger actions

* Code review by Arjun

* Code review by Arjun

* Code review by Omar
This commit is contained in:
Mark Towers
2022-09-03 23:39:23 +01:00
committed by GitHub
parent 8e74fe3b62
commit f39747d6a2
14 changed files with 741 additions and 151 deletions

View File

@@ -409,10 +409,10 @@ SPACE_KWARGS = [
{"nvec": [3, 2]}, # MultiDiscrete
{"n": 2}, # MultiBinary
{"max_length": 5}, # Text
# {"spaces": (Discrete(3), Discrete(2))}, # Tuple
# {"spaces": {"a": Discrete(3), "b": Discrete(2)}}, # Dict
# {"node_space": Discrete(4), "edge_space": Discrete(3)}, # Graph
# {"space": Discrete(4)}, # Sequence
{"spaces": (Discrete(3), Discrete(2))}, # Tuple
{"spaces": {"a": Discrete(3), "b": Discrete(2)}}, # Dict
{"node_space": Discrete(4), "edge_space": Discrete(3)}, # Graph
{"space": Discrete(4)}, # Sequence
]
assert len(SPACE_CLS) == len(SPACE_KWARGS)