Skip to content
Snippets Groups Projects
Commit 57ce4b61 authored by David Hammer's avatar David Hammer
Browse files

Hotfix from SPB: always nd.array the mask (can be VECTOR_BOOL)

parent c15bbc65
No related branches found
No related tags found
5 merge requests!59Interface CrystFEL with Karabo and allow automatic parameter tunning with rcrystfel,!53Train picker arbiter kernel,!52SFX hitfinder and geometry for correction addons,!47Draft: CrystFEL integration,!35Frame selection
...@@ -469,9 +469,11 @@ class ShmemTrainMatcher(TrainMatcher.TrainMatcher): ...@@ -469,9 +469,11 @@ class ShmemTrainMatcher(TrainMatcher.TrainMatcher):
new_sources_map = {} new_sources_map = {}
frame_selection_mask = None frame_selection_mask = None
if self._frame_selection_enabled and self._frame_selection_arbiter in sources: if self._frame_selection_enabled and self._frame_selection_arbiter in sources:
frame_selection_mask = sources[self._frame_selection_arbiter][0][ frame_selection_mask = np.array(
"data.dataFramePattern" sources[self._frame_selection_arbiter][0][
].astype(np.bool, copy=False) "data.dataFramePattern"
], copy=False
).astype(np.bool, copy=False),
if self._thread_pool is None: if self._thread_pool is None:
for source, (data, timestamp) in sources.items(): for source, (data, timestamp) in sources.items():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment