Fix: set local context to avoid errors resetting global start method (#…

…92) * Fix: set local context to avoid errors resetting global start method * Fixing trailing whitespace for code checks
plinder-org · Dec 31, 2024 · 5c1b90c · 5c1b90c
1 parent 7bcb4cc
commit 5c1b90c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml
@@ -4,7 +4,7 @@ on:
     tags:
       - 'v[0-9]+.[0-9]+.[0-9]+'
   workflow_dispatch:
-  
+
 permissions:
   id-token: write
   contents: read

diff --git a/src/plinder/eval/docking/write_scores.py b/src/plinder/eval/docking/write_scores.py
@@ -212,8 +212,8 @@ def score_test_set(
         )
         predictions = predictions[~predictions["exists"]]
     LOG.info(f"Running evaluation on {predictions.shape[0]} cases")
-    multiprocessing.set_start_method("spawn")
-    with multiprocessing.Pool(num_processes) as p:
+    ctx = multiprocessing.get_context("spawn")
+    with ctx.Pool(num_processes) as p:
         p.starmap(
             write_scores_as_json,
             [