KeyError in Training Splits Part

[gamzengineer]

Hi, I tried to running the command on Colab

!python create_splits.py --task dummy_mtl_concat --seed 1 --k 1

And I am getting this error:

Traceback (most recent call last):
File "create_splits.py", line 25, in
dataset = Generic_WSI_MTL_Dataset(csv_path = '/content/drive/MyDrive/Colab Notebooks/MTDL/TOAD/dataset_csv/dummy_dataset.csv',
File "/content/drive/My Drive/Colab Notebooks/MTDL/TOAD/datasets/dataset_mtl_concat.py", line 71, in init
slide_data = self.df_prep(slide_data, self.label_dicts, self.label_cols)
File "/content/drive/My Drive/Colab Notebooks/MTDL/TOAD/datasets/dataset_mtl_concat.py", line 133, in df_prep
data.at[i, 'label'] = label_dicts[0][key]
KeyError: 'Esophagogogastric'

Could anyone please explain why this is happening and how to resolve this error?