remove files before writing lib (custom kinfraglib)

kinfraglib/filters/retro.py

@@ -3,6 +3,7 @@
 """
 
 import multiprocessing as mp
+import os
 import pandas as pd
 import requests
 import redo
@@ -1387,6 +1388,13 @@ def save_fragment_library_to_sdfs(path_output, fragment_library_concat):
     path_output = Path(path_output)
     path_output.mkdir(parents=True, exist_ok=True)
 
+    subpockets = ["AP", "SE", "FP", "GA", "B1", "B2", "X"]
+    # remove all {subpocket}.sdf files before writing to avoid death files
+    for subpocket in subpockets:
+        # check if files {subpocket}.sdf exists
+        if (path_output / f"{subpocket}.sdf").exists():
+            os.remove(path_output / f"{subpocket}.sdf")
+
     for subpocket, fragments in fragment_library_concat.groupby("subpocket"):
         with Chem.SDWriter(str(path_output / f"{subpocket}.sdf")) as w:
             for mol in fragments.ROMol_original:

notebooks/custom_kinfraglib/1_4_custom_filters_pairwise_retrosynthesizability.ipynb

@@ -1245,8 +1245,8 @@
       "Checking if all fragment pairs were requested..\n",
       "All fragment pairs were requested.\n",
       "Done.\n",
-      "CPU times: user 9min 37s, sys: 4.41 s, total: 9min 41s\n",
-      "Wall time: 11min 50s\n"
+      "CPU times: user 12min 9s, sys: 6.03 s, total: 12min 15s\n",
+      "Wall time: 15min 43s\n"
      ]
     }
    ],