pyment/configurations/data_split_configuration.py

from typing import List

from pydantic import BaseModel


class DataSplitConfiguration(BaseModel):
    training_fraction: float
    stratification: List[str] = None