Try-except for capturing failure to start ray instances.

oncoray · Mar 27, 2024 · e11d6f6 · e11d6f6
1 parent 18a1c68
commit e11d6f6
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/mirp/utilities/parallel.py b/mirp/utilities/parallel.py
@@ -67,7 +67,15 @@ def ray_is_initialized():
 
 def ray_init(num_cpus):
     if RAY_AVAILABLE:
-        ray.init(num_cpus=num_cpus)
+        try:
+            ray.init(num_cpus=num_cpus)
+        except OSError as err:
+            warnings.warn(
+                f"Ray instances could not be started for parallel processing. Switching to sequential processing. "
+                f"{str(err)}",
+                UserWarning
+            )
+            pass
     else:
         warnings.warn(
             "The ray package was not found. Switching to sequential processing.",