updating overloads (#13)

Author: weinbe58

src/bloqade/geometry/dialects/grid/_interface.py

@@ -113,6 +113,10 @@ def get_ypos(grid: Grid[typing.Any, Ny]) -> ilist.IList[float, Ny]:
     ...
 
 
+@typing.overload
+def sub_grid(
+    grid: Grid, x_indices: ilist.IList[int, Nx], y_indices: ilist.IList[int, Ny]
+) -> Grid[Nx, Ny]: ...
 @typing.overload
 def sub_grid(
     grid: Grid, x_indices: ilist.IList[int, Nx], y_indices: list[int]