fix zip file path

GRAAL-Research · Nov 14, 2023 · a553da1 · a553da1
1 parent 8b93582
commit a553da1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)