From 09157ce36f9ef0c3129e051a14457d7950776513 Mon Sep 17 00:00:00 2001 From: s495728 Date: Sun, 9 Jun 2024 14:52:22 +0200 Subject: [PATCH] Change requirements --- requirements.txt | 5 +++-- src/service/dialog_policy.py | 4 +++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 75af938..232fa7f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ flair==0.13.1 conllu==4.5.3 pandas==1.5.3 -numpy==1.26.4 torch==2.3.0 -convlab==3.0.2a0 \ No newline at end of file +convlab==3.0.2a0 +numpy==1.24.4 +fuzzywuzzy==0.18.0 diff --git a/src/service/dialog_policy.py b/src/service/dialog_policy.py index cafa615..12eaacc 100644 --- a/src/service/dialog_policy.py +++ b/src/service/dialog_policy.py @@ -56,4 +56,6 @@ class SimpleRulePolicy(Policy): if domain in ["pizza", "drink"]: system_action[(domain, 'Recommend')].append(['Name', choice['name']]) if domain in ["size"]: - system_action[(domain, 'Recommend')].append(['Size', choice['size']]) \ No newline at end of file + system_action[(domain, 'Recommend')].append(['Size', choice['size']]) + +dialogPolicy = SimpleRulePolicy() \ No newline at end of file