From bdd78673b54f0086f405630b747b805078371e19 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Wed, 26 Jun 2024 11:14:38 +0200
Subject: [PATCH] [common.database] Hotfix for database check command
---
src/mednet/libs/common/scripts/database.py | 14 ++++++++++----
1 file changed, 10 insertions(+), 4 deletions(-)
diff --git a/src/mednet/libs/common/scripts/database.py b/src/mednet/libs/common/scripts/database.py
index 07c6f1f8..7bb561dc 100644
--- a/src/mednet/libs/common/scripts/database.py
+++ b/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}")
--
GitLab