From 57ce4b614b69f5cfee4fefde8a9a4c925075c98d Mon Sep 17 00:00:00 2001 From: David Hammer <dhammer@mailbox.org> Date: Tue, 21 Mar 2023 15:34:10 +0100 Subject: [PATCH] Hotfix from SPB: always nd.array the mask (can be VECTOR_BOOL) --- src/calng/ShmemTrainMatcher.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/calng/ShmemTrainMatcher.py b/src/calng/ShmemTrainMatcher.py index ae9fa539..279f65b4 100644 --- a/src/calng/ShmemTrainMatcher.py +++ b/src/calng/ShmemTrainMatcher.py @@ -469,9 +469,11 @@ class ShmemTrainMatcher(TrainMatcher.TrainMatcher): new_sources_map = {} frame_selection_mask = None if self._frame_selection_enabled and self._frame_selection_arbiter in sources: - frame_selection_mask = sources[self._frame_selection_arbiter][0][ - "data.dataFramePattern" - ].astype(np.bool, copy=False) + frame_selection_mask = np.array( + sources[self._frame_selection_arbiter][0][ + "data.dataFramePattern" + ], copy=False + ).astype(np.bool, copy=False), if self._thread_pool is None: for source, (data, timestamp) in sources.items(): -- GitLab