Skip to content

Commit 073f4c1

Browse files
committed
fix ConcatenateDataset to yield ItemsNotDefined, if some keys are duplicated
1 parent d7fe68b commit 073f4c1

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

lazy_dataset/core.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2703,6 +2703,12 @@ def ordered(self) -> bool:
27032703
return all(ds.ordered for ds in self.input_datasets)
27042704

27052705
def __iter__(self, with_key=False):
2706+
if with_key:
2707+
try:
2708+
self.keys()
2709+
except AssertionError:
2710+
raise _ItemsNotDefined(self.__class__.__name__) from None
2711+
27062712
for input_dataset in self.input_datasets:
27072713
if with_key:
27082714
iterable = input_dataset.__iter__(with_key=True)

0 commit comments

Comments
 (0)