add try catch

eth-easl · Oct 19, 2023 · a39bde9 · a39bde9
1 parent 8c554f4
commit a39bde9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/benchmark/cloc/data_generation.py b/benchmark/cloc/data_generation.py
@@ -31,8 +31,12 @@ def extract_single_zip(directory: str, zip_file: str) -> None:
 
     os.makedirs(output_dir, exist_ok=True)
 
-    with zipfile.ZipFile(zip_path, "r") as zip_ref:
-        zip_ref.extractall(output_dir)
+    try:
+        with zipfile.ZipFile(zip_path, "r") as zip_ref:
+            zip_ref.extractall(output_dir)
+    except Exception as e:
+        logger.error(f"Error while extracing file {zip_path}")
+        logger.error(e) 
 
 def setup_argparser() -> argparse.ArgumentParser:
     parser_ = argparse.ArgumentParser(description=f"CLOC Benchmark Storage Script")