60b6d7a3777d1e0c4b17bc954823951aac857fab,io/eolearn/io/sh_add.py,AddGeopediaFeature,execute,#AddGeopediaFeature#Any#,295

Before Change

        request_data = np.asarray(request.get_data())

        if eopatch.feature_exists(self.feature_type, self.feature_name):
            raster = eopatch.get_feature(self.feature_type, self.feature_name)
        else:
            raster = np.ones(dst_shape[1:3], dtype=self.raster_dtype) * self.no_data_val

        new_raster = self._reproject(eopatch, self._to_binary_mask(request_data[0]))

        // update raster
        raster[new_raster != 0] = new_raster[new_raster != 0]

After Change

        request_data, = np.asarray(request.get_data())

        if isinstance(self.raster_value, dict):
            raster = self._map_from_multiclass(eopatch, (height, width), request_data)
        elif isinstance(self.raster_value, (int, float)):
            raster = self._map_from_binaries(eopatch, (height, width), request_data)
        else:
            raise ValueError("Unsupported raster value type")

        if (self.feature_type in [FeatureType.MASK_TIMELESS]) and raster.ndim == 2:
            raster = raster[..., np.newaxis]

        eopatch.add_feature(self.feature_type, self.feature_name, raster)

Instances