Minor opt to cache tuple casting (#3894)

82455a41 · Aaron Gokaslan · GitHub · 2a7cb008 · 82455a41
Unverified Commit 82455a41 authored Apr 24, 2022 by Aaron Gokaslan Committed by GitHub Apr 24, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

include/pybind11/numpy.h include/pybind11/numpy.h +3 -2

No files found.
--- a/include/pybind11/numpy.h
+++ b/include/pybind11/numpy.h
@@ -647,8 +647,9 @@ private:
        for (auto field : attr("fields").attr("items")()) {
            auto spec = field.cast<tuple>();
            auto name = spec[0].cast<pybind11::str>();
-            auto format = spec[1].cast<tuple>()[0].cast<dtype>();
+            auto spec_fo = spec[1].cast<tuple>();
-            auto offset = spec[1].cast<tuple>()[1].cast<pybind11::int_>();
+            auto format = spec_fo[0].cast<dtype>();
+            auto offset = spec_fo[1].cast<pybind11::int_>();
            if ((len(name) == 0u) && format.kind() == 'V') {
                continue;
            }