Error on empty make_splits (#326)

Author: kozlov721

luxonis_ml/data/datasets/luxonis_dataset.py

@@ -1118,6 +1118,8 @@ def make_splits(
             if isinstance(splits, tuple):
                 ratios = splits
             elif isinstance(splits, dict):
+                if not splits:
+                    raise ValueError("Splits cannot be empty")
                 value = next(iter(splits.values()))
                 if isinstance(value, float):
                     ratios = splits  # type: ignore

tests/test_data/test_dataset.py

@@ -237,6 +237,9 @@ def generator(step: int = 15) -> DatasetIterator:
     with pytest.raises(ValueError, match="No new files"):
         dataset.make_splits()
 
+    with pytest.raises(ValueError, match="Splits cannot be empty"):
+        dataset.make_splits({})
+
     with pytest.raises(ValueError, match="Ratios must sum to 1.0"):
         dataset.make_splits((0.7, 0.1, 1))