Source code for rastervision.core.data.raster_transformer.cast_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.cast_transformer import ( # noqa
CastTransformer)
[docs]@register_config('cast_transformer')
class CastTransformerConfig(RasterTransformerConfig):
"""Configure a :class:`.CastTransformer`."""
to_dtype: str = Field(
...,
description='dtype to cast raster to. Must be a valid Numpy dtype '
'e.g. "uint8", "float32", etc.')