fix zipfile export

GRAAL-Research · Nov 14, 2023 · 0144238 · 0144238
1 parent a553da1
commit 0144238
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/training/evaluate_metrics.py b/src/training/evaluate_metrics.py
@@ -59,7 +59,7 @@
     url = "https://github.com/GRAAL-Research/MeaningBERT/releases/download/dependencies_model_release/LENS.zip"
     filehandle, _ = urllib.request.urlretrieve(url)
     with ZipFile(filehandle, "r") as zip_file:
-        zip_file.extractall(lens_model_ckpt, members=None, pwd=None)
+        zip_file.extractall(".", members=None, pwd=None)
 lens = LENS(lens_model_ckpt, rescale=True)
 
 coverage_model_ckpt = os.path.join("coverage_roberta.bin")
@@ -70,7 +70,7 @@
     )
     filehandle, _ = urllib.request.urlretrieve(url)
     with ZipFile(filehandle, "r") as zip_file:
-        zip_file.extractall(coverage_model_ckpt, members=None, pwd=None)
+        zip_file.extractall(".", members=None, pwd=None)
 
 coverage_kis = CoverageModel(model_file=coverage_model_ckpt)