Format checking of train.csv

[PradeepYLN]

I would like to confirm the processing I am doing correct or not. after running process_data.py by passing xBD directory and it generates train.csv and polygons. I looked at train.csv format and error are below when passing it to python damage_classification.py --train_data

,uuid,labels
0,9e483432-8681-4257-81db-335c382dbc32.png,0
1,99600a2d-588f-4980-98d8-325c1ea9c825.png,0
2,843424e5-f7c5-4af5-aff0-baddf6bc5836.png,0
3,0aa8779d-a857-435e-bc58-b55cc516971c.png,1

File "/damage_classifier/xView2_baseline/model/damage_classification.py", line 156, in train_model
    class_weights = compute_class_weight('balanced', np.unique(df['labels'].to_list()), df['labels'].to_list());
TypeError: compute_class_weight() takes 1 positional argument but 3 were given

can you please confirm the format of train.csv and cause of the error?