Restore get env if config exists

Denys88 · Feb 3, 2023 · 49edcfd · 49edcfd
1 parent eacfd0f
commit 49edcfd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/rl_games/common/player.py b/rl_games/common/player.py
@@ -24,8 +24,7 @@ def __init__(self, params):
  self.env_info = self.env.get_env_info()
  self.num_agents = 1
  else:
- # is there any way 'self.env' can be used in this scenario?
- # self.env = config.get('vec_env')
+ self.env = config.get('vec_env')
  self.num_agents = self.env_info['agents']
  self.value_size = self.env_info.get('value_size', 1)
  self.action_space = self.env_info['action_space']