From 21500b979d2da3e78319c05d8d1fbde87a152b80 Mon Sep 17 00:00:00 2001
From: Maximilian Schmidt <maximilian.schmidt@ims.uni-stuttgart.de>
Date: Tue, 20 Aug 2024 09:38:32 +0200
Subject: [PATCH] Fix runtime error
---
evoprompt/task/task.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/evoprompt/task/task.py b/evoprompt/task/task.py
index 8588373..1ba13ff 100644
--- a/evoprompt/task/task.py
+++ b/evoprompt/task/task.py
@@ -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
--
GitLab