Skip to content

Commit 74f44ff

Browse files
committed
fix(project_euler): fix var name and type hints
1 parent ab153b2 commit 74f44ff

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

project_euler/problem_060/sol1.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -175,17 +175,17 @@ def add_prime(primes: list[int]) -> list[int]:
175175
return primes
176176

177177

178-
def generate_primes(n: int, flag: int = 1) -> list[int]:
178+
def generate_primes(num_primes: int, flag: int = 1) -> list[int]:
179179
"""
180-
Ensure we have at least n primes in the selected list.
180+
Generates a list of the first num_primes primes based on their modulo 3 value.
181181
182182
>>> generate_primes(5, 1)
183183
[3, 7, 13, 19, 31]
184184
>>> generate_primes(5, 2)
185185
[3, 5, 11, 17, 23]
186186
"""
187187
primes = prime_mod_3_is_1_list if flag == 1 else prime_mod_3_is_2_list
188-
while len(primes) < n:
188+
while len(primes) < num_primes:
189189
primes = add_prime(primes)
190190
return primes
191191

@@ -212,7 +212,7 @@ def solution(
212212
raise ValueError("flag must be either 1 or 2")
213213
primes = generate_primes(prime_limit, flag)
214214

215-
def search(chain):
215+
def search(chain: tuple) -> tuple[int, ...] | None:
216216
"""
217217
Recursive backtracking search to find a valid set of primes.
218218
A threshold is used to ensure we don't exceed the smallest sum.

0 commit comments

Comments
 (0)