Skip to content

Commit

Permalink
fix pyright
Browse files Browse the repository at this point in the history
  • Loading branch information
Chuck Tang committed Jun 3, 2024
1 parent 79ad163 commit c3f74e5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/data/test_icl_datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def run_test(
for i, e in enumerate(evaluators):
batch = next(e.dataloader.dataloader.__iter__())
# Check that the dataloader is the correct length for the first task.
if i == 0:
if i == 0 and hasattr(e.dataloader.dataloader, '__len__'):
if eval_drop_last:
assert len(
e.dataloader.dataloader,
Expand Down

0 comments on commit c3f74e5

Please sign in to comment.