diff --git a/shared/database/utils/get_dataset.py b/shared/database/utils/get_dataset.py
index 5440df1..0c823cc 100755
--- a/shared/database/utils/get_dataset.py
+++ b/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 []