Change run_stardis() function signature to allow manual input of `a…

…tom_data`
smokestacklightnin · Nov 12, 2023 · 8da20c8 · 8da20c8
1 parent afcc0ca
commit 8da20c8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/stardis/base.py b/stardis/base.py
@@ -18,7 +18,7 @@
 schema = os.path.join(base_dir, "config_schema.yml")
 
 
-def run_stardis(config_fname, tracing_lambdas_or_nus):
+def run_stardis(config_fname, tracing_lambdas_or_nus, atom_data=None):
     """
     Runs a STARDIS simulation.
 
@@ -69,6 +69,8 @@ def run_stardis(config_fname, tracing_lambdas_or_nus):
             )
             + 1,
         )
+        if atom_data == None
+        else atom_data
     )
     # plasma
     stellar_plasma = create_stellar_plasma(stellar_model, adata)