from __future__ import annotations 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.""" return []