Move OUTSIDE_EJECTA to PacketStatus ENUM

tardis-sn · Jul 30, 2024 · edd7f43 · edd7f43
1 parent a542b37
commit edd7f43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/tardis/transport/montecarlo/r_packet.py b/tardis/transport/montecarlo/r_packet.py
@@ -26,11 +26,7 @@ class PacketStatus(IntEnum):
     EMITTED = 1
     REABSORBED = 2
     ADIABATIC_COOLING = 4
-
-
-class PacketEjectaStatus(IntEnum):
-    OUT_EJECTA = -1
-    IN_EJECTA = 1
+    OUTSIDE_EJECTA = -1
 
 
 rpacket_spec = [

diff --git a/tardis/transport/montecarlo/single_packet_loop.py b/tardis/transport/montecarlo/single_packet_loop.py
@@ -22,7 +22,6 @@
 from tardis.transport.montecarlo.r_packet import (
     InteractionType,
     PacketStatus,
-    PacketEjectaStatus,
 )
 from tardis.transport.montecarlo.r_packet_transport import (
     move_packet_across_shell_boundary,
@@ -87,7 +86,7 @@ def single_packet_loop(
 
     rpacket_tracker.track(r_packet)
     rpacket_tracker.track_boundary_interaction(
-        PacketEjectaStatus.OUT_EJECTA, r_packet.current_shell_id
+        PacketStatus.OUTSIDE_EJECTA, r_packet.current_shell_id
     )
 
     # this part of the code is temporary and will be better incorporated
@@ -178,7 +177,7 @@ def single_packet_loop(
             temp_next_shell_id = (
                 r_packet.current_shell_id
                 if r_packet.status == PacketStatus.IN_PROCESS
-                else PacketEjectaStatus.OUT_EJECTA
+                else PacketStatus.OUTSIDE_EJECTA
             )
             rpacket_tracker.track_boundary_interaction(
                 temp_current_shell_id, temp_next_shell_id