Skip to content

Commit a53710b

Browse files
committed
Update __init__.py
1 parent 85c4f7e commit a53710b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

artisatomic/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1530,7 +1530,9 @@ def write_output_files(
15301530

15311531
if unused_upsilon_transitions:
15321532
dfupsilon_only_transitions = pl.DataFrame(
1533-
list(unused_upsilon_transitions), schema={"lowerlevel": pl.Int64, "upperlevel": pl.Int64}, orient="row"
1533+
list(unused_upsilon_transitions),
1534+
schema=(("lowerlevel", pl.Int64), ("upperlevel", pl.Int64)),
1535+
orient="row",
15341536
).with_columns(A=0.0)
15351537
for id_lower, id_upper in dfupsilon_only_transitions[["lowerlevel", "upperlevel"]].iter_rows(named=False):
15361538
namefrom = dfenergylevels_ion["levelname"][id_upper]

0 commit comments

Comments
 (0)