Fix imports

tardis-sn · Dec 3, 2024 · 9a900f6 · 9a900f6
1 parent daa6539
commit 9a900f6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/tardis/plasma/equilibrium/rates/__init__.py b/tardis/plasma/equilibrium/rates/__init__.py
@@ -2,9 +2,18 @@
     UpsilonCMFGENSolver,
     UpsilonRegemorterSolver,
 )
+from tardis.plasma.equilibrium.rates.collisional_ionization_rates import (
+    CollisionalIonizationSolver,
+)
+from tardis.plasma.equilibrium.rates.collisional_ionization_strengths import (
+    CollisionalIonizationSeaton,
+)
 from tardis.plasma.equilibrium.rates.collisional_rates import (
     ThermalCollisionalRateSolver,
 )
+from tardis.plasma.equilibrium.rates.photoionization_rates import (
+    PhotoionizationRateSolver,
+)
 from tardis.plasma.equilibrium.rates.radiative_rates import (
     RadiativeRatesSolver,
 )
diff --git a/tardis/plasma/equilibrium/rates/collisional_ionization_rates.py b/tardis/plasma/equilibrium/rates/collisional_ionization_rates.py
@@ -1,4 +1,6 @@
-from collisional_ionization_strengths import CollisionalIonizationSeaton
+from tardis.plasma.equilibrium.rates.collisional_ionization_strengths import (
+    CollisionalIonizationSeaton,
+)
 
 
 class CollisionalIonizationSolver: