AxisError: axis 4 is out of bounds for array of dimension 4

t0=time.time()

hist_vanilla = model_vanilla.fit_generator(
    SeismicSequence(
        labeled_data,
        train_data["Xline"].values,
        train_data["Time"].values,
        train_data["Class"].values,
        patch_size,
        batch_size,
        1),
    steps_per_epoch=steps,
    validation_data = SeismicSequence(
        labeled_data,
        test_data["Xline"].values,
        test_data["Time"].values,
        test_data["Class"].values,
        patch_size,
        batch_size,
        1),
    validation_steps = len(test_samples)//batch_size,
    epochs = epochs,
    verbose = 0,
    callbacks = callbacklist)
     

print('--> Training for Waldeland CNN took {:.1f} sec'.format(time.time()-t0)) #Thanks to aadm 

AxisError                                 Traceback (most recent call last)
<ipython-input-52-8828ecedd241> in <module>
     22     epochs = epochs,
     23     verbose = 0,
---> 24     callbacks = callbacklist)
     25 
     26 

5 frames
<__array_function__ internals> in expand_dims(*args, **kwargs)

/usr/local/lib/python3.7/dist-packages/numpy/core/numeric.py in <listcomp>(.0)
   1383             pass
   1384     # Going via an iterator directly is slower than via list comprehension.
-> 1385     axis = tuple([normalize_axis_index(ax, ndim, argname) for ax in axis])
   1386     if not allow_duplicate and len(set(axis)) != len(axis):
   1387         if argname:

AxisError: axis 4 is out of bounds for array of dimension 4

I'm try to excude this code but I got this erorr showing that the axis no 4 is out of bands for the array of dimension 4, befor that it gives me erorr on the callbackslist, can any one help please?

source https://stackoverflow.com/questions/74593855/axiserror-axis-4-is-out-of-bounds-for-array-of-dimension-4