"""Definition of get_dataset function."""

from typing import Literal

from pymongo.collection import Collection


def get_dataset(
    collection: Collection,
    type: Literal['train', 'test', 'validation'],
) -> list[str]:
    """Get list of data names for the corresponding type."""
    assert isinstance(collection, Collection)
    assert isinstance(type, str)
    assert type in ['train', 'test', 'validation']
    return []