[common.database] Hotfix for database check command

src/mednet/libs/common/scripts/database.py

@@ -75,10 +75,16 @@ def check(entry_point_group, fold, limit):  # numpydoc ignore=PR01
             for i, batch in enumerate(loader):
                 if loader_limit == 0:
                     break
-                logger.info(
-                    f"{batch[1]['name'][0]}: "
-                    f"{[s for s in batch[0][0].shape]}@{batch[0][0].dtype}",
-                )
+                if isinstance(batch[0], dict):
+                    logger.info(
+                        f"{batch[1]['name'][0]}: "
+                        f"{[s for s in batch[0]['image'][0].shape]}@{batch[0]['image'][0].dtype}",
+                    )
+                else:
+                    logger.info(
+                        f"{batch[1]['name'][0]}: "
+                        f"{[s for s in batch[0][0].shape]}@{batch[0][0].dtype}",
+                    )
                 loader_limit -= 1
         except Exception:
             logger.exception(f"Unable to load batch {i} in dataset {k}")