Fix bug for adding simulations in series

pynbody · Oct 11, 2023 · 55fff96 · 55fff96
1 parent 1238b8e
commit 55fff96
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tangos/tools/add_simulation.py b/tangos/tools/add_simulation.py
@@ -14,9 +14,12 @@ def __init__(self, simulation_output, session=None, renumber=True, parallel=Fals
         """:type simulation_output tangos.simulation_outputs.HandlerBase"""
         self.simulation_output = simulation_output
         if session is None:
-            # if running in parallel, creating a session for the first time may trigger a race condition e.g. if the
-            # database doesn't exist yet
-            with pt.ExclusiveLock("creating_database"):
+            if parallel:
+                # if running in parallel, creating a session for the first time may trigger a race condition e.g. if the
+                # database doesn't exist yet
+                with pt.ExclusiveLock("creating_database"):
+                    session = core.get_default_session()
+            else:
                 session = core.get_default_session()
         self.session = session
         self.min_halo_particles = config.min_halo_particles