saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-28 22:56:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

implemented set_env method

Browse source
This commit is contained in:
Noah Dormann 2019-12-05 09:11:30 +01:00
parent 8460bfe397
commit 4b1bab7f85
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
torchy_baselines/common/base_class.py
View file

@ -169,10 +169,20 @@ class BaseRLModel(object):
def set_env(self, env):
"""
Checks the validity of the environment, and if it is coherent, set it as the current environment.
checked parameters:
- observation_space
- action_space
:param env: (Gym Environment) The environment for learning a policy
"""
pass
if self.observation_space != env.observation_space:
raise ValueError("The given environment has a observation_space that doesn't fit the current model")
if self.action_space != env.action_space:
raise ValueError("The given environment has a action_space that doesn't fit the current model")
# if all fits save new env
self.env = env
def get_parameter_list(self):
"""