Native_Enum

pybind11-stubgen needs location hint for default arguments
that are enums, even if defined early enough.

Author: ax3l

.github/update_stub.sh

@@ -13,6 +13,6 @@ set -eu -o pipefail
 # we are in the source directory, .github/
 this_dir=$(cd $(dirname $0) && pwd)
 
-pybind11-stubgen --exit-code -o ${this_dir}/../src/ amrex.space1d
-pybind11-stubgen --exit-code -o ${this_dir}/../src/ amrex.space2d
-pybind11-stubgen --exit-code -o ${this_dir}/../src/ amrex.space3d
+pybind11-stubgen --exit-code --enum-class-locations="GrowthStrategy:amrex.space3d" -o ${this_dir}/../src/ amrex.space1d
+pybind11-stubgen --exit-code --enum-class-locations="GrowthStrategy:amrex.space3d" -o ${this_dir}/../src/ amrex.space2d
+pybind11-stubgen --exit-code --enum-class-locations="GrowthStrategy:amrex.space3d" -o ${this_dir}/../src/ amrex.space3d

src/Base/PODVector.cpp

@@ -150,11 +150,14 @@ void make_PODVector(py::module &m, std::string typestr)
 #endif
 }
 
-void init_PODVector(py::module& m) {
-    py::enum_<GrowthStrategy>(m, "GrowthStrategy")
+void init_PODVector(py::module& m)
+{
+    py::native_enum<GrowthStrategy>(m, "GrowthStrategy", "enum.IntEnum")
         .value("Poisson", GrowthStrategy::Poisson)
         .value("Exact", GrowthStrategy::Exact)
         .value("Geometric", GrowthStrategy::Geometric)
+        .export_values()
+        .finalize()
     ;
 
     make_PODVector<ParticleReal> (m, "real");