#%%

import pickle

with open('/Users/morteza/Downloads/participants_data/participants_data_v2021/full_track/sub01/WB.pkl', 'rb') as f:
  df = pickle.load(f)

df['train'].shape