From 8b98e6986f20edea1f7aba699e04b0239407326b Mon Sep 17 00:00:00 2001
From: Dipam Chakraborty <dipamc77@gmail.com>
Date: Fri, 10 Nov 2023 11:49:45 +0530
Subject: [PATCH] fix: no need to remove last dialogue
---
agents/bart_agent.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/agents/bart_agent.py b/agents/bart_agent.py
index 894f060..36864f5 100644
--- a/agents/bart_agent.py
+++ b/agents/bart_agent.py
@@ -50,7 +50,7 @@ class BARTResponseAgent(object):
persona = [tokenize(line.strip()) for line in conversation['persona B']]
# partner = [tokenize(line.strip()) for line in conversation['persona A']]
partner = [] # Baseline not trained with the partner personaj
- history = [tokenize(line['text'].strip()) for line in conversation['dialogue'][:-1]]
+ history = [tokenize(line['text'].strip()) for line in conversation['dialogue']]
return persona, partner, history
def prepare_tensors(self, conversation):
--
GitLab