from typing import Any

from ...base import Case, Dataset, Model
from ...metrics.example_selection import SmallNoiseCheck


class ContinuityCase(Case):
    def __init__(
        self, ds: Dataset, noisy_ds: Dataset, model: Model, *args: Any, **kwargs: Any
    ) -> None:
        super().__init__(ds, model, *args, **kwargs)
        self.name = "continuity"
        self._metric_objs["small_noise_check"] = SmallNoiseCheck(ds, noisy_ds, model)
