From 8da20c87a226461769ebc493dadfc890d7b233cf Mon Sep 17 00:00:00 2001
From: William Black <125844868+smokestacklightnin@users.noreply.github.com>
Date: Sat, 11 Nov 2023 23:07:02 -0300
Subject: [PATCH] Change `run_stardis()` function signature to allow manual
 input of `atom_data`

---
 stardis/base.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/stardis/base.py b/stardis/base.py
index 284260e8..a15ae038 100644
--- 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)