#  Copyright © 2025 Emmi AI GmbH. All rights reserved.

import logging

from noether.core.schemas.dataset import DatasetBaseConfig, DatasetSplitIDs
from noether.data.datasets.cfd.caeml.ahmedml.split import AhmedMLDefaultSplitIDs
from noether.data.datasets.cfd.caeml.dataset import CAEMLDataset

logger = logging.getLogger(__name__)


class AhmedMLDataset(CAEMLDataset):
    """
    Dataset implementation for AhmedML CFD simulations.

    Args:
        dataset_config: Configuration for the dataset.
    """

    def __init__(
        self,
        dataset_config: DatasetBaseConfig,
    ):
        """
        Initialize the AhmedML dataset.

        Args:
            dataset_config: Configuration for the dataset.

        """
        super().__init__(dataset_config=dataset_config, dataset_name=self.get_dataset_splits.DATASET_NAME)  # type: ignore[arg-type]

    @property
    def get_dataset_splits(self) -> DatasetSplitIDs:
        return AhmedMLDefaultSplitIDs()
