From 2d8928f2aedeafbc5ff4b727821d99d7138bd87e Mon Sep 17 00:00:00 2001
From: Arec <jamgochian95@gmail.com>
Date: Thu, 21 Oct 2021 06:29:43 -0700
Subject: [PATCH] updating data processing scripts to output to the correct
 location

---
 generate_demos.sh      | 4 +---
 src/data/__init__.py   | 2 +-
 src/data/data_utils.py | 2 +-
 src/data/expert.py     | 2 +-
 4 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/generate_demos.sh b/generate_demos.sh
index 447068a..9900f5e 100755
--- a/generate_demos.sh
+++ b/generate_demos.sh
@@ -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 ../
\ No newline at end of file
+python -m src.data.expert --locs='[DR_USA_Roundabout_FT]' 
\ No newline at end of file
diff --git a/src/data/__init__.py b/src/data/__init__.py
index 49d78bc..a134dbc 100644
--- a/src/data/__init__.py
+++ b/src/data/__init__.py
@@ -1 +1 @@
-from src.data.expert_trajectories import demonstrations, load_experts, process_experts 
\ No newline at end of file
+from src.data.expert import demonstrations, load_experts, process_experts 
\ No newline at end of file
diff --git a/src/data/data_utils.py b/src/data/data_utils.py
index b14ec3a..0bcf9d4 100644
--- a/src/data/data_utils.py
+++ b/src/data/data_utils.py
@@ -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
 
diff --git a/src/data/expert.py b/src/data/expert.py
index fb6b754..401531f 100644
--- a/src/data/expert.py
+++ b/src/data/expert.py
@@ -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