Update SIR_model.py
Browse files- SIR_model.py +1 -1
SIR_model.py
CHANGED
|
@@ -223,7 +223,7 @@ def main(total_population, initial_infected, initial_recovered, beta, gamma, mod
|
|
| 223 |
anim = FuncAnimation(fig2, animate, frames=times, interval=50, blit=True)
|
| 224 |
plt.show()
|
| 225 |
|
| 226 |
-
return [fig1,
|
| 227 |
|
| 228 |
elif mode=="graph":
|
| 229 |
|
|
|
|
| 223 |
anim = FuncAnimation(fig2, animate, frames=times, interval=50, blit=True)
|
| 224 |
plt.show()
|
| 225 |
|
| 226 |
+
return [fig1, anim]
|
| 227 |
|
| 228 |
elif mode=="graph":
|
| 229 |
|