fix nopython in total_xs helper

Author: nglaser3

mcdc/transport/physics/interface.py

@@ -51,14 +51,13 @@ def total_xs(particle_container, simulation, data):
     """
     particle = particle_container[0]
     if particle["particle_type"] == PARTICLE_NEUTRON:
-        module = neutron
-        type_total = NEUTRON_REACTION_TOTAL
+        type = NEUTRON_REACTION_TOTAL
+        return neutron.macro_xs(type, particle_container, simulation, data)
     elif particle["particle_type"] == PARTICLE_ELECTRON:
         module = electron
-        type_total = ELECTRON_REACTION_TOTAL
-    else:
-        return 0.0
-    return module.macro_xs(type_total, particle_container, simulation, data)
+        type = ELECTRON_REACTION_TOTAL
+        return neutron.macro_xs(type, particle_container, simulation, data)
+    return 0.0
 
 
 @njit