Update Stub Files

ax3l · May 29, 2024 · d4d409b · d4d409b
1 parent 9424e69
commit d4d409b
Show file tree

Hide file tree

Showing 6 changed files with 414 additions and 12 deletions.
diff --git a/src/amrex/space1d/__init__.pyi b/src/amrex/space1d/__init__.pyi
@@ -493,4 +493,4 @@ __author__: str = (
     "Axel Huebl, Ryan T. Sandberg, Shreyas Ananthan, David P. Grote, Revathi Jambunathan, Edoardo Zoni, Remi Lehe, Andrew Myers, Weiqun Zhang"
 )
 __license__: str = "BSD-3-Clause-LBNL"
-__version__: str = "24.05"
+__version__: str = "24.05-32-g28b010126a1b"
diff --git a/src/amrex/space1d/amrex_1d_pybind/__init__.pyi b/src/amrex/space1d/amrex_1d_pybind/__init__.pyi
@@ -371,7 +371,12 @@ class Array4_cdouble:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: complex,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -506,7 +511,12 @@ class Array4_cdouble_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[complex]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -655,7 +665,12 @@ class Array4_cfloat:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: complex,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -790,7 +805,12 @@ class Array4_cfloat_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[complex]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -939,7 +959,12 @@ class Array4_double:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: float,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -1074,7 +1099,12 @@ class Array4_double_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.float64]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -1223,7 +1253,12 @@ class Array4_float:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: float,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -1358,7 +1393,12 @@ class Array4_float_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.float32]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -1507,7 +1547,12 @@ class Array4_int:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -1642,7 +1687,12 @@ class Array4_int_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.int32]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -1791,7 +1841,12 @@ class Array4_long:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -1926,7 +1981,12 @@ class Array4_long_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.int64]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -2075,7 +2135,12 @@ class Array4_longdouble:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: float,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -2210,7 +2275,12 @@ class Array4_longdouble_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.longdouble]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -2359,7 +2429,12 @@ class Array4_longlong:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -2494,7 +2569,12 @@ class Array4_longlong_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.int64]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -2643,7 +2723,12 @@ class Array4_short:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -2778,7 +2863,12 @@ class Array4_short_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.int16]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -2927,7 +3017,12 @@ class Array4_uint:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -3062,7 +3157,12 @@ class Array4_uint_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.uint32]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -3211,7 +3311,12 @@ class Array4_ulong:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -3346,7 +3451,12 @@ class Array4_ulong_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.uint64]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -3495,7 +3605,12 @@ class Array4_ulonglong:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -3630,7 +3745,12 @@ class Array4_ulonglong_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.uint64]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -3779,7 +3899,12 @@ class Array4_ushort:
         arg0: typing.Annotated[list[int], pybind11_stubgen.typing_ext.FixedSize(3)],
         arg1: int,
     ) -> None: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -3914,7 +4039,12 @@ class Array4_ushort_const:
     @typing.overload
     def __init__(self, arg0: numpy.ndarray[numpy.uint16]) -> None: ...
     def __repr__(self) -> str: ...
+    @typing.overload
     def contains(self, arg0: int, arg1: int, arg2: int) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: IntVect) -> bool: ...
+    @typing.overload
+    def contains(self, arg0: Dim3) -> bool: ...
     def to_cupy(self, copy=False, order="F"):
         """
 
@@ -4499,7 +4629,11 @@ class Box:
         Return the length of the Box in given direction.
         """
 
-    def make_slab(self, direction: int, slab_index: int) -> Box: ...
+    def make_slab(self, direction: int, slab_index: int) -> Box:
+        """
+        Flatten the box in one direction.
+        """
+
     def normalize(self) -> None: ...
     def numPts(self) -> int:
         """
@@ -4628,7 +4762,7 @@ class BoxArray:
     def size(self) -> int: ...
 
 class Config:
-    amrex_version: typing.ClassVar[str] = "24.05"
+    amrex_version: typing.ClassVar[str] = "24.05-32-g28b010126a1b"
     gpu_backend = None
     have_gpu: typing.ClassVar[bool] = False
     have_mpi: typing.ClassVar[bool] = True
@@ -16293,4 +16427,4 @@ __author__: str = (
     "Axel Huebl, Ryan T. Sandberg, Shreyas Ananthan, David P. Grote, Revathi Jambunathan, Edoardo Zoni, Remi Lehe, Andrew Myers, Weiqun Zhang"
 )
 __license__: str = "BSD-3-Clause-LBNL"
-__version__: str = "24.05"
+__version__: str = "24.05-32-g28b010126a1b"
diff --git a/src/amrex/space2d/__init__.pyi b/src/amrex/space2d/__init__.pyi
@@ -541,4 +541,4 @@ __author__: str = (
     "Axel Huebl, Ryan T. Sandberg, Shreyas Ananthan, David P. Grote, Revathi Jambunathan, Edoardo Zoni, Remi Lehe, Andrew Myers, Weiqun Zhang"
 )
 __license__: str = "BSD-3-Clause-LBNL"
-__version__: str = "24.05"
+__version__: str = "24.05-32-g28b010126a1b"