add arg to the config_schema.yml file for gzipped and hook up in run_…

…stardis (#116) * add arg to the config_schema.yml file for gzipped and hook up in run_stardis * remove comment about adding gzipped arg
tardis-sn · Aug 11, 2023 · dc79d40 · dc79d40
1 parent d6bde59
commit dc79d40
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/stardis/base.py b/stardis/base.py
@@ -48,10 +48,9 @@ def run_stardis(config_fname, tracing_lambdas_or_nus):
     if config.model.type == "marcs":
         from stardis.io.model.marcs import read_marcs_model
 
-        # FIX THIS BY ADDING AN OPTIONAL GZIPPED ARGUMENT TO THE CONFIG
         raw_marcs_model = read_marcs_model(
-            config.model.fname, gzipped=False
-        )  # Need to add gzipped to config
+            config.model.fname, gzipped=config.model.gzipped
+        )
         stellar_model = raw_marcs_model.to_stellar_model(
             adata, final_atomic_number=config.model.final_atomic_number
         )

diff --git a/stardis/config_schema.yml b/stardis/config_schema.yml
@@ -16,10 +16,14 @@ properties:
                 - marcs
             fname:
                 type: string
+            gzipped:
+                type: boolean
+                default: false            
             final_atomic_number:
                 type: number
                 multipleOf: 1
                 default: 30
+
         required:
         - type
         - fname