Adjusted Callable type hint

This commit is contained in:
Fabian
2022-10-24 09:24:12 +02:00
parent ed645c2fbe
commit 5ebd4225cc
3 changed files with 8 additions and 7 deletions
+2 -1
View File
@@ -1,4 +1,5 @@
from itertools import chain
from types import FunctionType
from typing import Tuple, Type, Union, Optional
import gym
@@ -120,7 +121,7 @@ def test_replanning_time(mp_type: str, env_wrap: Tuple[str, Type[RawInterfaceWra
{'basis_generator_type': 'rbf'}, seed=SEED)
assert env.do_replanning
assert env.replanning_schedule
assert callable(env.replanning_schedule)
# This also verifies we are not adding the TimeAwareObservationWrapper twice
assert env.observation_space == env_step.observation_space