From a62b8179fdad2a4640f84a02539a03358d78d18c Mon Sep 17 00:00:00 2001
From: zqwerty <zhuq96@hotmail.com>
Date: Mon, 27 Jun 2022 13:34:13 +0800
Subject: [PATCH] update multiwoz21 database.py
---
data/unified_datasets/multiwoz21/database.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/data/unified_datasets/multiwoz21/database.py b/data/unified_datasets/multiwoz21/database.py
index 0d9f0f4e..e79d5535 100644
--- a/data/unified_datasets/multiwoz21/database.py
+++ b/data/unified_datasets/multiwoz21/database.py
@@ -5,13 +5,14 @@ from fuzzywuzzy import fuzz
from itertools import chain
from zipfile import ZipFile
from copy import deepcopy
-from convlab.util.unified_datasets_util import BaseDatabase
+from convlab.util.unified_datasets_util import BaseDatabase, download_unified_datasets
class Database(BaseDatabase):
def __init__(self):
"""extract data.zip and load the database."""
- archive = ZipFile(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data.zip'))
+ data_path = download_unified_datasets('multiwoz21', 'data.zip', os.path.dirname(os.path.abspath(__file__)))
+ archive = ZipFile(data_path)
self.domains = ['restaurant', 'hotel', 'attraction', 'train', 'hospital', 'police']
self.dbs = {}
for domain in self.domains:
--
GitLab