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

Merge branch 'add-demo-for-evaluation' into 'master'

Fix runtime error

See merge request !4
parents 719e2ffa 21500b97
No related branches found
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