diff --git a/mushroom_rl/core/multiprocess_environment.py b/mushroom_rl/core/multiprocess_environment.py index ce7633a9..6f566851 100644 --- a/mushroom_rl/core/multiprocess_environment.py +++ b/mushroom_rl/core/multiprocess_environment.py @@ -66,6 +66,7 @@ def __init__(self, env_class, *args, n_envs=-1, use_generator=False, **kwargs): **kwargs: keyword arguments to set to the constructor or to the generator; """ + assert env_class is not None, "Environment class requires not installed module." assert n_envs > 1 or n_envs == -1 if n_envs == -1: