From a553da1a7cb453c4d21ef857787b18187b5706a8 Mon Sep 17 00:00:00 2001 From: davebulaval Date: Mon, 13 Nov 2023 19:17:58 -0500 Subject: [PATCH] fix zip file path --- src/training/evaluate_metrics.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/training/evaluate_metrics.py b/src/training/evaluate_metrics.py index 0ee5f9c..68d38cc 100644 --- a/src/training/evaluate_metrics.py +++ b/src/training/evaluate_metrics.py @@ -58,7 +58,7 @@ if not os.path.exists(lens_model_ckpt): url = "https://github.com/GRAAL-Research/MeaningBERT/releases/download/dependencies_model_release/LENS.zip" filehandle, _ = urllib.request.urlretrieve(url) - with ZipFile(".", "r") as zip_file: + with ZipFile(filehandle, "r") as zip_file: zip_file.extractall(lens_model_ckpt, members=None, pwd=None) lens = LENS(lens_model_ckpt, rescale=True) @@ -69,7 +69,7 @@ "coverage_roberta.bin" ) filehandle, _ = urllib.request.urlretrieve(url) - with ZipFile(".", "r") as zip_file: + with ZipFile(filehandle, "r") as zip_file: zip_file.extractall(coverage_model_ckpt, members=None, pwd=None) coverage_kis = CoverageModel(model_file=coverage_model_ckpt)