From 048568a4c42ea7483d733abc59987e66adb15744 Mon Sep 17 00:00:00 2001
From: Silin Gao <54292608+Silin159@users.noreply.github.com>
Date: Mon, 26 Feb 2024 19:15:26 +0100
Subject: [PATCH] Add files via upload

---
 BART_Rel_Pred/baseline/dataset.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/BART_Rel_Pred/baseline/dataset.py b/BART_Rel_Pred/baseline/dataset.py
index 1e63778..2a86430 100644
--- a/BART_Rel_Pred/baseline/dataset.py
+++ b/BART_Rel_Pred/baseline/dataset.py
@@ -16,7 +16,13 @@
 
 logger = logging.getLogger(__name__)
 
-ADD_TOKENS_VALUES = ["<fact_sep>", "<rel_bos>", "<rel_eos>"]
+ADD_TOKENS_VALUES = ["<utter_sep>", "<past>", "<center>", "<future>", "<rel_bos>", "<rel_eos>",
+                     "personx", "persony", "personz", "<fact_sep>", "<eos_fact>",
+                     "<atlocation>", "<capableof>", "<causes>", "<desires>",
+                     "<hasproperty>", "<hassubevent>", "<hinderedby>", "<isafter>",
+                     "<isbefore>", "<madeupof>", "<notdesires>", "<objectuse>",
+                     "<oeffect>", "<oreact>", "<owant>", "<xattr>", "<xeffect>",
+                     "<xintent>", "<xneed>", "<xreact>", "<xreason>", "<xwant>"]
 
 
 class BaseDataset(torch.utils.data.Dataset):