From 21b63b9616f3a709724462997fa16993aa7e2865 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Niklas=20K=C3=B6hnecke?=
 <niklas.koehnecke@ext.aleph-alpha.com>
Date: Thu, 30 May 2024 13:19:50 +0200
Subject: [PATCH] fix: ommited rebase changes

---
 tests/evaluation/evaluation/test_argilla_evaluator.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/evaluation/evaluation/test_argilla_evaluator.py b/tests/evaluation/evaluation/test_argilla_evaluator.py
index 78ecec58f..9c5526adb 100644
--- a/tests/evaluation/evaluation/test_argilla_evaluator.py
+++ b/tests/evaluation/evaluation/test_argilla_evaluator.py
@@ -42,6 +42,15 @@ class StubArgillaClient(ArgillaClient):
     _datasets: dict[str, list[RecordData]] = {}
     _score = 3.0
 
+    def create_dataset(
+        self,
+        workspace_id: str,
+        dataset_name: str,
+        fields: Sequence[Field],
+        questions: Sequence[Question],
+    ) -> str:
+        return self.ensure_dataset_exists(workspace_id, dataset_name, fields, questions)
+
     def ensure_dataset_exists(
         self,
         workspace_id: str,