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

from typing import Optional

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: Optional[float] = Field( 0.0, description=('Turn all NaN values into this value.'))
[docs] def build(self): return NanTransformer(to_value=self.to_value)