Update seir_model.py

seir_model.py

@@ -1,7 +1,7 @@
 import numpy as np
 
 def p_exposed(params, substep, state_history, previous_state):
-  signal = previous_state["S"]*params["infection_rate"]*previous_state["I"]/N
+  signal = previous_state["S"]*params["infection_rate"]*previous_state["I"]/(previous_state["S"]+previous_state["E"]+previous_state["I"]+previous_state["R"])
   return {"delta_s":np.ceil(signal)}
   
 def p_infected(params, substep, state_history, previous_state):