Added new task to registry

KennethEnevoldsen · Aug 23, 2024 · 1677dda · 1677dda
1 parent 5e9a0b5
commit 1677dda
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/seb/registered_tasks/danish.py b/src/seb/registered_tasks/danish.py
@@ -101,3 +101,14 @@ def create_twitterhjerne() -> Task:
     task.domain = ["social"]
     task.task_subtypes = ["Question-answering"]
     return task
+
+
+@tasks.register("HistoricalDanishClustering")
+def create_historical() -> Task:
+    from seb.mteb_tasks import HistoricalDanishClustering
+
+    task = MTEBTask(HistoricalDanishClustering())
+    task.name = "HistoricalDanishClustering"
+    task.domain = ["poetry", "fiction"]
+    task.task_subtypes = ["Thematic Clustering"]
+    return task