shared/database/utils/get_dataset.py

@@ -1,4 +1,4 @@
-from __future__ import annotations
+"""Definition of get_dataset function."""
 
 from typing import Literal
 
@@ -10,4 +10,7 @@ def get_dataset(
     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 []