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)