Add conditional in case of prior ray.init

lanl · Oct 8, 2024 · 050fb87 · 050fb87
1 parent 3772c50
commit 050fb87
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/scripts/ct_datagen_foam2.py b/examples/scripts/ct_datagen_foam2.py
@@ -19,7 +19,10 @@
 import logging
 import ray
 
-ray.init(logging_level=logging.ERROR)  # need to call init before jax import: ray-project/ray#44087
+if not ray.is_initialized():
+    ray.init(
+        logging_level=logging.ERROR
+    )  # need to call init before jax import: ray-project/ray#44087
 
 from scico import plot
 from scico.flax.examples import load_ct_data