Add functions to __init__.py

ai2cm · Oct 13, 2023 · 2374ac7 · 2374ac7
1 parent c2c4f52
commit 2374ac7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/wrapper/shield/wrapper/__init__.py b/wrapper/shield/wrapper/__init__.py
@@ -5,6 +5,9 @@
     initialize,
     step,
     step_dynamics,
+    step_radiation,
+    step_pre_radiation,
+    step_post_radiation_physics,
     step_physics,
     save_intermediate_restart_if_enabled,
     save_fortran_restart,