Source code for rastervision.core.data.label_source.label_source_config

from typing import TYPE_CHECKING, Optional

from rastervision.pipeline.config import Config, register_config

if TYPE_CHECKING:
    from rastervision.core.box import Box
    from rastervision.core.data import (ClassConfig, CRSTransformer,
                                        LabelSource, SceneConfig)
    from rastervision.core.rv_pipeline import RVPipelineConfig


@register_config('label_source')
class LabelSourceConfig(Config):
    """Configure a :class:`.LabelSource`."""

    def build(self,
              class_config: 'ClassConfig',
              crs_transformer: 'CRSTransformer',
              bbox: Optional['Box'] = None,
              tmp_dir: Optional[str] = None) -> 'LabelSource':
        raise NotImplementedError()

    def update(self,
               pipeline: Optional['RVPipelineConfig'] = None,
               scene: Optional['SceneConfig'] = None) -> None:
        pass