14 lines
294 B
Python
Executable File
14 lines
294 B
Python
Executable File
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 []
|