My code is runtime error. Help me fix that
import numpy as np
def sigma_mod(n):
if n==1:
return 0
result = 0
for i in range(1, n):
for j in range(1, n + i):
s = i + j
if n + s < 1000:
k = np.arange(1, n + s, dtype=np.int64)
denom = i + j + k
safe_denom = np.where(denom == 0, 1, denom)
mod_vals = (i * j * k) % safe_denom
result += np.sum(mod_vals)
else:
for k in range(1, n + s):
denom = s + k
if denom != 0:
result += (i * j * k) % denom
return result
if name == "main":
n = int(input())
if n > 0:
numbers = []
for i in range(n):
num = int(input())
if num >= 0:
numbers.append(num)
else:
break
results = [sigma_mod(num) for num in numbers]
for result in results:
print(result)