error in train.py file in root at line number 100

[rajeshtva]

Good afternoon sir.

there is an error saying undefined variable CLASS at line number 100 in the train.py file

---------------------------------------------------------------------------

NameError                                 Traceback (most recent call last)

<ipython-input-9-66b3304bbf9c> in <module>()
     94     X_FLAIR_val = to_uint8(get_data(FLAIR_val))[None,None,...]
     95     X_IR_val = IR_to_uint8(get_data(IR_val))[None,None,...]
---> 96     y_val = np.array(get_data(segm_val)==CLASS).astype(np.uint8)[None,...]
     97     print("STARTING TRAINING...")
     98     model_ = model()

NameError: name 'CLASS' is not defined

i am running this code in colab so. please ignore line number here