"""
Ma + Olaniru et al. (2023) dataset.
"""

from scTimeBench.shared.dataset.base import BaseDataset, ObservationColumns
import scanpy as sc


class MaOlaniruDataset(BaseDataset):
    def _load_data(self):
        """
        Load the Ma + Olaniru et al. dataset.
        """
        print("Loading Ma +Olaniru et al. dataset...")
        # read from the dataset data_path
        data_path = self.dataset_dict["data_path"]
        self.data = sc.read_h5ad(data_path)

        self.data.obs = self.data.obs.rename(
            columns={
                "timepoint": ObservationColumns.TIMEPOINT.value,
            }
        )
        self.data.obs[ObservationColumns.CELL_TYPE.value] = "unknown"
        print(" Ma + Olaniru et al. dataset loaded successfully.")

        print(
            f"Timepoints: {self.data.obs[ObservationColumns.TIMEPOINT.value].unique()}"
        )
        print("No celltypes available for this dataset as of yet.")
        print("Assigned dummy cell type: unknown")
