renamed module

shared/docstore/src/get_dataset.py

@@ -0,0 +1,16 @@
+"""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 []