From 0f94f565ddee278e246b5b397c4ac49b8c20e0a2 Mon Sep 17 00:00:00 2001 From: sukhijab Date: Thu, 30 Nov 2023 16:25:21 +0100 Subject: [PATCH] changed directory of loading for car2 --- experiments/data_provider.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/experiments/data_provider.py b/experiments/data_provider.py index 621d935..6ad987a 100644 --- a/experiments/data_provider.py +++ b/experiments/data_provider.py @@ -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]