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

fixing expert data combiner and adding feasibility checking with all options to util.collisions

Browse Source
This commit is contained in:
Arec
2021-10-21 06:59:21 -07:00
parent 2d8928f2ae
commit 06785236d4
3 changed files with 161 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/data/expert.py
View File

@@ -109,11 +109,12 @@ def load_experts(expert_files):
Returns:
transitions (list): list of combined expert episode transitions
"""
transitions = []
trajectories = []
for file in tqdm(expert_files):
with open(file, "rb") as f:
trajectories = pickle.load(f)
transitions = transitions + rollout.flatten_trajectories(trajectories)
new_trajectories = pickle.load(f)
trajectories += new_trajectories
transitions = rollout.flatten_trajectories(trajectories)
return transitions
def demonstrations(expert='NormalizedIntersimpleExpert', env='NRasterizedIncrementingAgent', path=None, min_timesteps=None, min_episodes=None, video=False, env_args={}, policy_args={}):