17 lines
431 B
Python
Executable File
17 lines
431 B
Python
Executable File
"""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 []
|