Implemented Extend action for NumPy array

TUDelft-CNS-ATM · Sep 11, 2023 · 2a821ae · 2a821ae
1 parent f267075
commit 2a821ae
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/bluesky/network/sharedstate.py b/bluesky/network/sharedstate.py
@@ -144,13 +144,14 @@ def receive(action, data):
  container.append(item)
 
  elif ctx.action == ActionType.Extend:
- # TODO: other types? (ndarray, ...)
  for key, item in data.items():
  container = getattr(store, key, None)
  if container is None:
  setattr(store, key, item)
- else:
+ elif isinstance(container, list):
  container.extend(item)
+ elif isinstance(container, np.ndarray):
+ setattr(store, key, np.concatenate([container, item]))
 
  elif ctx.action == ActionType.Replace:
  vars(store).update(data)