Skip to content
Snippets Groups Projects

Fix runtime error

Merged Max Kimmich requested to merge add-demo-for-evaluation into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
+ 1
1
@@ -328,7 +328,7 @@ class Task(metaclass=ABCMeta):
lambda _, idx: idx in samples_ids, with_indices=True
)
# 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
)
return demonstration_samples, remaining_dataset
Loading