Update __init__.py

artis-mcrt · Dec 18, 2024 · 7da0e17 · 7da0e17
1 parent 9b796be
commit 7da0e17
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/artisatomic/__init__.py b/artisatomic/__init__.py
@@ -1485,7 +1485,11 @@ def write_output_files(
 
         log_and_print(flog, f"\n===========> Z={atomic_number} {ionstr} output:")
 
-        dfenergylevels_ion = leveltuples_to_pldataframe(energy_levels[i]).with_row_index(name="levelid")
+        dfenergylevels_ion = (
+            leveltuples_to_pldataframe(energy_levels[i])
+            .with_row_index(name="levelid")
+            .with_columns(pl.col("levelid").cast(pl.Int64))
+        )
 
         dftransitions_ion = dftransitions_allions[i]