Skip to content
Snippets Groups Projects
Commit 21500b97 authored by Max Kimmich's avatar Max Kimmich
Browse files

Fix runtime error

parent 2b2251c4
Branches add-demo-for-evaluation
No related tags found
1 merge request!4Fix runtime error
...@@ -328,7 +328,7 @@ class Task(metaclass=ABCMeta): ...@@ -328,7 +328,7 @@ class Task(metaclass=ABCMeta):
lambda _, idx: idx in samples_ids, with_indices=True lambda _, idx: idx in samples_ids, with_indices=True
) )
# remove demonstration samples from validation set # remove demonstration samples from validation set
remaining_dataset = self.dataset.filter( remaining_dataset = dataset.filter(
lambda _, idx: idx not in samples_ids, with_indices=True lambda _, idx: idx not in samples_ids, with_indices=True
) )
return demonstration_samples, remaining_dataset return demonstration_samples, remaining_dataset
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment