diff --git a/tardis/montecarlo/montecarlo_transport_state.py b/tardis/montecarlo/montecarlo_transport_state.py index 441f26a3f79..22ab1181bef 100644 --- a/tardis/montecarlo/montecarlo_transport_state.py +++ b/tardis/montecarlo/montecarlo_transport_state.py @@ -365,9 +365,7 @@ def virt_packet_initial_mus(self): @property def virt_packet_last_interaction_in_nu(self): try: - return u.Quantity( - self.vpacket_tracker.last_interaction_in_nu, u.erg - ) + return u.Quantity(self.vpacket_tracker.last_interaction_in_nu, u.Hz) except AttributeError: warnings.warn( "MontecarloTransport.virt_packet_last_interaction_in_nu:" @@ -381,7 +379,7 @@ def virt_packet_last_interaction_in_nu(self): @property def virt_packet_last_interaction_type(self): try: - return u.Quantity(self.vpacket_tracker.last_interaction_type, u.erg) + return self.vpacket_tracker.last_interaction_type except AttributeError: warnings.warn( "MontecarloTransport.virt_packet_last_interaction_type:" @@ -395,9 +393,7 @@ def virt_packet_last_interaction_type(self): @property def virt_packet_last_line_interaction_in_id(self): try: - return u.Quantity( - self.vpacket_tracker.last_interaction_in_id, u.erg - ) + return self.vpacket_tracker.last_interaction_in_id except AttributeError: warnings.warn( "MontecarloTransport.virt_packet_last_line_interaction_in_id:" @@ -411,9 +407,7 @@ def virt_packet_last_line_interaction_in_id(self): @property def virt_packet_last_line_interaction_out_id(self): try: - return u.Quantity( - self.vpacket_tracker.last_interaction_out_id, u.erg - ) + return self.vpacket_tracker.last_interaction_out_id except AttributeError: warnings.warn( "MontecarloTransport.virt_packet_last_line_interaction_out_id:" @@ -427,9 +421,7 @@ def virt_packet_last_line_interaction_out_id(self): @property def virt_packet_last_line_interaction_shell_id(self): try: - return u.Quantity( - self.vpacket_tracker.last_interaction_shell_id, u.erg - ) + return self.vpacket_tracker.last_interaction_shell_id except AttributeError: warnings.warn( "MontecarloTransport.virt_packet_last_line_interaction_shell_id:"