import os
import pickle

def data_preprocess(deg=15):
    traj_root = "/data1/tpz/nwm-main/data_splits/airvln_16/test"
    traj_file_path = os.path.join(traj_root, f"rollout_turn_{deg}deg.pkl")
    with open(traj_file_path, 'rb') as f:
        traj_data = pickle.load(f)
    
    print(traj_data[29])

if __name__ == "__main__":
    data_preprocess(deg=15)