huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve custom env support

Browse Source
This commit is contained in:
ebuehrle
2021-11-03 17:48:27 +01:00
parent ba850eadbb
commit 2718f52941
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scratch/etienne/intersimple/data/expert.py
View File

@@ -1,10 +1,11 @@
from intersim.envs.intersimple import Intersimple, InfoFilter from intersim.envs.intersimple import Intersimple, InfoFilter
from stable_baselines3.common.policies import BasePolicy from stable_baselines3.common.policies import BasePolicy
import gym import gym
import intersim.envs.intersimple from intersim.envs.intersimple import *
import imitation.data.rollout as rollout import imitation.data.rollout as rollout
from stable_baselines3.common.vec_env.dummy_vec_env import DummyVecEnv from stable_baselines3.common.vec_env.dummy_vec_env import DummyVecEnv
from imitation.data.wrappers import RolloutInfoWrapper from imitation.data.wrappers import RolloutInfoWrapper
from gail.envs import NRasterizedRouteSpeedRandomAgentLocation
class IntersimExpert(BasePolicy): class IntersimExpert(BasePolicy):
@@ -105,7 +106,7 @@ def demonstrations(expert='NormalizedIntersimpleExpert', env='NRasterizedRandomA
python -m intersimple.expert <flags> python -m intersimple.expert <flags>
""" """
Env = intersim.envs.intersimple.__dict__[env] Env = globals()[env]
Expert = globals()[expert] Expert = globals()[expert]
env = Env(**env_args) env = Env(**env_args)