Add correct annotation

GridTools · Jan 17, 2024 · f9bd983 · f9bd983
1 parent 4a7d62a
commit f9bd983
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gt4py/next/iterator/transforms/power_unrolling.py b/src/gt4py/next/iterator/transforms/power_unrolling.py
@@ -33,7 +33,7 @@ def _is_power_call(
     )
 
 
-def _compute_integer_power_of_two(exp) -> int:
+def _compute_integer_power_of_two(exp: int) -> int:
     return math.floor(math.log2(exp))