exception

[hanriiver]

case 1) N_WEEKS_PASSED > N_WEEKS_TOTAL

Since inertia = max(N_WEEKS_TOTAL - N_WEEKS_PASSED, 0), it becomes 0, causing dropoff to be fixed at 2.0.
This eliminates its role in adjusting the growth rate.

solution
Limit N_WEEKS_PASSED so that it does not exceed N_WEEKS_TOTAL.

case 2) DISTRIBUTION = [] empty

In for i in range(1, len(DISTRIBUTION)), if the list is empty, len(DISTRIBUTION) = 0, so the loop does not execute.

I wrote this for fun