Source code for etna.transforms.utils

import reprlib
from typing import Iterable
from typing import Optional

from etna.datasets.utils import inverse_transform_target_components  # noqa: F401
from etna.datasets.utils import match_target_quantiles  # noqa: F401

[docs]def check_new_segments(transform_segments: Iterable[str], fit_segments: Optional[Iterable[str]]): """Check if there are any new segments that weren't present during training.""" if fit_segments is None: raise ValueError("Transform is not fitted!") new_segments = set(transform_segments) - set(fit_segments) if len(new_segments) > 0: raise NotImplementedError( f"This transform can't process segments that weren't present on train data: {reprlib.repr(new_segments)}" )