Create source length zeroed arrays from yielded `None` fsp history

2022-02-02 08:06:41 -05:00 · 2022-02-02 08:06:41 -05:00 · f3289c1977
parent 4e96dd09e3
commit f3289c1977
1 changed files with 7 additions and 1 deletions
--- a/piker/fsp/_engine.py
+++ b/piker/fsp/_engine.py
@ -135,11 +135,17 @@ async def fsp_compute(
                output = history_output[key]

                if history is None:
+
+                    if output is None:
+                        length = len(src.array)
+                    else:
+                        length = len(output)
+
                    # using the first output, determine
                    # the length of the struct-array that
                    # will be pushed to shm.
                    history = np.zeros(
-                        len(output),
+                        length,
                        dtype=dst.array.dtype
                    )