Skip to content

Commit c4038e3

Browse files
Removed list / tuple case for datamodule str method
1 parent 28ee099 commit c4038e3

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

src/lightning/pytorch/core/datamodule.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -261,11 +261,6 @@ def __str__(self) -> str:
261261
datasets_info.append(f"{attr_name}, dataset size={len(attr)}")
262262
else:
263263
datasets_info.append(f"{attr_name}, dataset size=Unavailable")
264-
elif isinstance(attr, (list, tuple)) and all(isinstance(item, Dataset) for item in attr):
265-
if all(hasattr(item, "__len__") for item in attr):
266-
datasets_info.append(f"{attr_name}, dataset size={[len(ds) for ds in attr]}")
267-
else:
268-
datasets_info.append(f"{attr_name}, dataset size=Unavailable")
269264

270265
if not datasets_info:
271266
return "No datasets are set up."

0 commit comments

Comments
 (0)