updated to not get vulture error
This commit is contained in:
@@ -1,4 +1,4 @@
|
|||||||
from __future__ import annotations
|
"""Definition of get_dataset function."""
|
||||||
|
|
||||||
from typing import Literal
|
from typing import Literal
|
||||||
|
|
||||||
@@ -10,4 +10,7 @@ def get_dataset(
|
|||||||
type: Literal['train', 'test', 'validation'],
|
type: Literal['train', 'test', 'validation'],
|
||||||
) -> list[str]:
|
) -> list[str]:
|
||||||
"""Get list of data names for the corresponding type."""
|
"""Get list of data names for the corresponding type."""
|
||||||
|
assert isinstance(collection, Collection)
|
||||||
|
assert isinstance(type, str)
|
||||||
|
assert type in ['train', 'test', 'validation']
|
||||||
return []
|
return []
|
||||||
|
|||||||
Reference in New Issue
Block a user