From bdd0dfff10649d8c56d942cd1641add6d04a6513 Mon Sep 17 00:00:00 2001 From: Brian Bjarke Jensen Date: Sat, 13 Jul 2024 22:36:27 +0200 Subject: [PATCH] updated to not get vulture error --- shared/database/utils/get_dataset.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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 []