updated to not get vulture error

This commit is contained in:
Brian Bjarke Jensen
2024-07-13 22:36:27 +02:00
parent e913bb28cc
commit bdd0dfff10
+4 -1
View File
@@ -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 []