From eabe0ed7bc4092f1f756b0f038377d3a434d3ac8 Mon Sep 17 00:00:00 2001
From: svlandeg <svlandeg@github.com>
Date: Fri, 19 Jan 2024 18:29:45 +0100
Subject: [PATCH] typo fix

---
 tutorials/llm_clinical_trials/scripts/trial_task.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tutorials/llm_clinical_trials/scripts/trial_task.py b/tutorials/llm_clinical_trials/scripts/trial_task.py
index 704285ca..27933079 100644
--- a/tutorials/llm_clinical_trials/scripts/trial_task.py
+++ b/tutorials/llm_clinical_trials/scripts/trial_task.py
@@ -15,7 +15,7 @@
 Patient group: <name>
 Number of patients in the group: <number>
 Treatment drug or substance: <drug>
-Treatment dose: <drug>
+Treatment dose: <dose>
 Treatment frequency of administration: <frequency>
 Treatment duration: <duration>
 Outcome: <outcome>
@@ -114,9 +114,9 @@ def parse_responses(
             matcher.add("Patient_Group", [nlp.make_doc(text) for text in patient_groups])
             matcher.add("Patient_Number", [nlp.make_doc(text) for text in patient_numbers])
             matcher.add("Treatment_Drug", [nlp.make_doc(text) for text in drugs])
-            matcher.add("Dose", [nlp.make_doc(text) for text in doses])
-            matcher.add("Frequency", [nlp.make_doc(text) for text in frequencies])
-            matcher.add("Duration", [nlp.make_doc(text) for text in durations])
+            matcher.add("Treatment_Dose", [nlp.make_doc(text) for text in doses])
+            matcher.add("Treatment_Frequency", [nlp.make_doc(text) for text in frequencies])
+            matcher.add("Treatment_Duration", [nlp.make_doc(text) for text in durations])
             matcher.add("Outcome", [nlp.make_doc(text) for text in outcomes])
 
             matches = matcher(doc, as_spans=True)