fix: catch missing changes

scikit-hep · Sep 27, 2023 · 03b777d · 03b777d
1 parent 98ab307
commit 03b777d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/awkward/forms/recordform.py b/src/awkward/forms/recordform.py
@@ -311,5 +311,6 @@ def __setstate__(self, state):
     def _expected_from_buffers(
         self, getkey: Callable[[Form, str], str], recursive: bool
     ) -> Iterator[tuple[str, np.dtype]]:
-        for content in self._contents:
-            yield from content._expected_from_buffers(getkey)
+        if recursive:
+            for content in self._contents:
+                yield from content._expected_from_buffers(getkey, recursive)
diff --git a/src/awkward/forms/unionform.py b/src/awkward/forms/unionform.py
@@ -299,5 +299,6 @@ def _expected_from_buffers(
     ) -> Iterator[tuple[str, np.dtype]]:
         yield (getkey(self, "tags"), index_to_dtype[self._tags])
         yield (getkey(self, "index"), index_to_dtype[self._index])
-        for content in self._contents:
-            yield from content._expected_from_buffers(getkey)
+        if recursive:
+            for content in self._contents:
+                yield from content._expected_from_buffers(getkey, recursive)