fix: add check for image dimensions during loading (#15)

* fix: add check for image dimensions during loading

closes #14

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* refactor: refactor to comply with ruff

* docs(load_data): add Exception to function docstring

---------

Co-authored-by: Lukas Nepelius <p42940@fhooe.at>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: lukasbehammer <lukas.behammer@fh-wels.at>

Author: 4 people

src/viqa/utils/loading.py

@@ -655,7 +655,8 @@ def load_data(
     ------
     ValueError
         If input type is not supported \n
-        If ``data_range=None`` and ``normalize=True``
+        If ``data_range=None`` and ``normalize=True`` \n
+        If number of dimensions not 2 or 3
 
     Warns
     -----
@@ -738,6 +739,21 @@ def load_data(
                 "Input type not supported"
             )  # Raise exception if input type is not supported
 
+    if batch:
+        # TODO: Deprecate in version 4.0.0
+        for img_num, img in enumerate(img_arr):
+            if len(img.shape) > 3 or len(img.shape) < 2:
+                raise ValueError(
+                    f"Number of dimensions of input image should be 2 or 3, got "
+                    f"{len(img.shape)} for image number {img_num} in img_arr."
+                )
+    elif not isinstance(img_arr, list):
+        if len(img_arr.shape) > 3 or len(img_arr.shape) < 2:
+            raise ValueError(
+                f"Number of dimensions of input image should be 2 or 3, got "
+                f"{len(img_arr.shape)}."
+            )
+
     # Normalize data
     if normalize and data_range:
         # TODO: Deprecate in version 4.0.0