Source code for rastervision.core.data.raster_transformer.nan_transformer_config

from rastervision.pipeline.config import register_config, Field
from rastervision.core.data.raster_transformer.raster_transformer_config import (  # noqa
    RasterTransformerConfig)
from rastervision.core.data.raster_transformer.nan_transformer import (  # noqa
    NanTransformer)


[docs]@register_config('nan_transformer')
class NanTransformerConfig(RasterTransformerConfig):
    """Configure a :class:`.NanTransformer`."""

    to_value: float | None = Field(
        0.0, description=('Turn all NaN values into this value.'))

[docs]    def build(self, channel_order: list[int] | None = None):
        return NanTransformer(to_value=self.to_value)