add error reporting in dataclass_from_flat_list

glucauze · Aug 14, 2023 · 17db6ea · 17db6ea
1 parent 0a9d992
commit 17db6ea
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/faceswaplab_utils/ui_utils.py b/scripts/faceswaplab_utils/ui_utils.py
@@ -14,6 +14,10 @@ def dataclass_from_flat_list(cls: type, values: Tuple[Any, ...]) -> Any:
             init_values[field.name] = field.type(*inner_values)
             idx += len(inner_values)
         else:
+            if idx >= len(values):
+                raise IndexError(
+                    f"Expected more values for dataclass {cls}. Current index: {idx}, values length: {len(values)}"
+                )
             value = values[idx]
             init_values[field.name] = value
             idx += 1