Prime number Python homework

[Yoriz]

from itertools import count will keep returning values forever.
for number in count(): has no way of ending
while len(primes) < n: only happens once

The following code Removes the while loop and gives the for loop a way out.

from itertools import count
 
 
def is_prime(number):
    if number <= 1:
        return False
    for i in range(2, number):
        if number % i == 0:
            return False;
 
    return True
 

 
def get_primes(n):
    primes = []
    for number in count():
        if is_prime(number):
            primes.append(number)
        if len(primes) == n:
            return primes


print(get_primes(1))

Output:
[2]