From 43f43c0415d0b9619d432e48a69c2b3bab9efb27 Mon Sep 17 00:00:00 2001 From: Jeffrey Reep Date: Mon, 16 Dec 2024 14:40:29 -1000 Subject: [PATCH] undo change to util --- fiasco/tests/test_ion.py | 2 +- fiasco/util/util.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/fiasco/tests/test_ion.py b/fiasco/tests/test_ion.py index 3eeeb72..7bb76be 100644 --- a/fiasco/tests/test_ion.py +++ b/fiasco/tests/test_ion.py @@ -168,7 +168,7 @@ def test_ionization_fraction_setter(ion): assert u.allclose(ion.ionization_fraction, 0.0) ion.ionization_fraction = 0.1 assert u.allclose(ion.ionization_fraction, 0.1) - + def test_formation_temperature(ion): assert ion.formation_temperature == ion.temperature[np.argmax(ion.ionization_fraction)] diff --git a/fiasco/util/util.py b/fiasco/util/util.py index d0c8585..c7f6de0 100644 --- a/fiasco/util/util.py +++ b/fiasco/util/util.py @@ -113,9 +113,8 @@ def walk_sub_dir(subdir): return subdir_files - non_ion_subdirs = ['abundance', 'ip', 'continuum', 'dem'] + non_ion_subdirs = ['abundance', 'ip', 'ioneq', 'continuum', 'dem'] all_files = {f'{sd}_files': walk_sub_dir(sd) for sd in non_ion_subdirs} - all_files['ionization_files'] = walk_sub_dir('ioneq') all_files['ion_files'] = ion_files return all_files