reuse the abstract

ARBML · Jun 22, 2024 · c82574d · c82574d
1 parent a372062
commit c82574d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/embeddings_utils.py b/utils/embeddings_utils.py
@@ -62,4 +62,4 @@ def compute_embeddings(texts: List[str], model_name: str) -> List[List[float]]:
 
 
 def build_dataset_prompt(dataset: Dict[str, Union[str, int]]) -> str:
-    return f"{dataset['Name']} {dataset['Description']} {dataset['Tasks']}"
+    return f"{dataset['Name']} {dataset['Description']} {dataset['Tasks']} {dataset['Abstract']}"