Skip to content

Commit

Permalink
changed directory of loading for car2
Browse files Browse the repository at this point in the history
  • Loading branch information
sukhijab committed Nov 30, 2023
1 parent 54fb611 commit 0f94f56
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions experiments/data_provider.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,8 +199,7 @@ def get_rccar_recorded_data_new(encode_angle: bool = True, skip_first_n_points:
recordings_dir = [os.path.join(DATA_DIR, 'recordings_rc_car_v1')]
elif car_id == 2:
num_train_traj = 10
recordings_dir = [os.path.join(DATA_DIR, 'recordings_rc_car_v2'),
os.path.join(DATA_DIR, 'recordings_rc_car_v3')]
recordings_dir = [os.path.join(DATA_DIR, 'recordings_rc_car_v4')]
else:
raise ValueError(f"Unknown car id {car_id}")
files = [sorted(glob.glob(rd + '/*.pickle')) for rd in recordings_dir]
Expand Down

0 comments on commit 0f94f56

Please sign in to comment.