ValueError: During execution of task EuclideanNormTask

I am using a python scripts template instance and when executing the following workflow:

workflow = LinearWorkflow(
       add_data,
       add_clm,
       NDVI,
       NDWI,
       NORM,
       add_sh_valmask,
       count_val_sh,
       *land_cover_task_array,
       concatenate,
       filter_task,
       linear_interp,
       erosion,
       spatial_sampling,
       save_s2)

I got this error:

When I used this code instead of the standard from eolearn.features import EuclideanNormTask , I think it got solved because. Does it mean that the built-in EuclideanNormTask is bugged?

class EuclideanNorm(EOTask):   
   """
   The tasks calculates Euclidian Norm of all bands within an array:
   norm = sqrt(sum_i Bi**2),
   where Bi are the individual bands within user-specified feature array.
   """
   def __init__(self, feature_name, in_feature_name):
       self.feature_name = feature_name
       self.in_feature_name = in_feature_name
   
   def execute(self, eopatch):
       arr = eopatch.data[self.in_feature_name]
       norm = np.sqrt(np.sum(arr**2, axis=-1))
       
       eopatch.add_feature(FeatureType.DATA, self.feature_name, norm[..., np.newaxis])
       return eopatch