From 03ecb17ef635cd6585c38f6045e30bd02a5fff60 Mon Sep 17 00:00:00 2001 From: "Raffin, Antonin" Date: Thu, 5 Dec 2019 14:41:39 +0100 Subject: [PATCH] Update error message --- torchy_baselines/common/base_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py index 4a08917..0bd1fae 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -193,7 +193,7 @@ class BaseRLModel(object): :param env: (Gym Environment) The environment for learning a policy """ if self.check_env(env, self.observation_space, self.action_space) is False: - raise ValueError("Given environment is not compatible with model") + raise ValueError("The given environment is not compatible with model: observation and action spaces do not match") # it must be coherent now # if it is not a VecEnv, make it a VecEnv if not isinstance(env, VecEnv):