fix: ommited rebase changes

Aleph-Alpha · May 31, 2024 · bf87edc · bf87edc
1 parent 61fee6f
commit bf87edc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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,