Fix type annotations for integer_partition.py TheAlgorithms#4052 (The…

…Algorithms#4689)
changhaibo0323 · Aug 30, 2021 · 3acca3d · 3acca3d
1 parent 8e5c353
commit 3acca3d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dynamic_programming/integer_partition.py b/dynamic_programming/integer_partition.py
@@ -6,8 +6,8 @@
 """
 
 
-def partition(m):
-    memo = [[0 for _ in range(m)] for _ in range(m + 1)]
+def partition(m: int) -> int:
+    memo: list[list[int]] = [[0 for _ in range(m)] for _ in range(m + 1)]
     for i in range(m + 1):
         memo[i][0] = 1