updating data processing scripts to output to the correct location

This commit is contained in:
Arec
2021-10-21 06:29:43 -07:00
parent 05b31092f4
commit 2d8928f2ae
4 changed files with 4 additions and 6 deletions

View File

@@ -6,6 +6,4 @@
# expert_class:str='NormalizedIntersimpleExpert',
# expert_args:dict={mu:0.001}):
cd ./src/data
python -m expert --locs='[DR_USA_Roundabout_FT]'
cd ../
python -m src.data.expert --locs='[DR_USA_Roundabout_FT]'

View File

@@ -1 +1 @@
from src.data.expert_trajectories import demonstrations, load_experts, process_experts
from src.data.expert import demonstrations, load_experts, process_experts

View File

@@ -1,7 +1,7 @@
import torch
from torch.utils.data import Dataset
import numpy as np
from src.expert_data import load_expert_data
from src.data.expert_data import load_expert_data
import os
opj = os.path.join

View File

@@ -100,7 +100,7 @@ def save_video(env, expert):
env.render()
env.close()
def load_experts(expert_files=[]):
def load_experts(expert_files):
"""
Load expert trajectories from files and combine their transitions into a single RB