Modify DATA_START_ROW

tardis-sn · Mar 12, 2024 · 567fa84 · 567fa84
1 parent fdb51b8
commit 567fa84
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tardis/io/model/readers/stella.py b/tardis/io/model/readers/stella.py
@@ -21,8 +21,7 @@ class STELLAModel:
     ("\s+total mass\s+(\d+\.\d+E[+-]\d+)\s+\d+\.\d+E[+-]\d+", "total_mass"),
 ]
 
-DATA_START_ROW = 6
-
+DATA_START_ROW = 5
 COLUMN_WITH_UNIT_RE = re.compile("(.+)\s+\((.+)\)")
 
 
@@ -46,9 +45,8 @@ def read_stella_model(fname):
         for i, line in enumerate(fh):
             if i < len(HEADER_RE_STR):
                 header_re_match = header_re[i].match(line)
-
                 metadata[HEADER_RE_STR[i][1]] = header_re_match.group(1)
-            elif i + 1 == DATA_START_ROW:
+            elif i == DATA_START_ROW:
                 if "mass of cell" in line:
                     column_names_raw = re.split(r"\s{3,}", line.strip())
                     break