Primes Test with AWS Lambda

primes.py

import json
import numpy as np

def compute_primes_up_to(n):
    # https://stackoverflow.com/questions/2068372/fastest-way-to-list-all-primes-below-n-in-python/3035188#3035188
    """ Input n>=6, Returns a array of primes, 2 <= p < n """
    sieve = np.ones(int(n/3) + (n%6==2), dtype=np.bool)
    sieve[0] = False
    for i in range(int(int(n**0.5)/3+1)):
        if sieve[i]:
            k=3*i+1|1
            sieve[   int((k*k)/3)      ::2*k] = False
            sieve[int((k*k+4*k-2*k*(i&1))/3)::2*k] = False
    return np.r_[2,3,((3*np.nonzero(sieve)[0]+1)|1)]
    
def lambda_handler(event, context):
    for i in range(1000):
        primes = compute_primes_up_to(1000000)
        print("Found %s primes" % len(primes))
    return {
        'OK': 'OK'
    }