Skip to content
Prev Previous commit
Next Next commit
Update loreal_poc/dataloaders/base.py
  • Loading branch information
rabah-khalek authored Jan 4, 2024
commit a9cac74a59fa4bc7b927be4795e5d3419b08adc1
2 changes: 1 addition & 1 deletion loreal_poc/dataloaders/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def __iter__(self):
self.idx = 0
return self

def flat_len(self):
def flat_len(self) -> int:
return len(self.idx_sampler)

def __len__(self) -> int:
Expand Down