From 6705ee04538e49a995649008276a295b4f7cb649 Mon Sep 17 00:00:00 2001 From: Eric Denovellis Date: Tue, 16 Jan 2024 13:02:08 -0800 Subject: [PATCH] Handle numpy arrays (#766) --- src/spyglass/lfp/lfp_electrode.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/spyglass/lfp/lfp_electrode.py b/src/spyglass/lfp/lfp_electrode.py index 4d2d75269..0b683a3da 100644 --- a/src/spyglass/lfp/lfp_electrode.py +++ b/src/spyglass/lfp/lfp_electrode.py @@ -1,4 +1,5 @@ import datajoint as dj +from numpy import ndarray from spyglass.common.common_ephys import Electrode from spyglass.common.common_session import Session # noqa: F401 @@ -48,6 +49,9 @@ def create_lfp_electrode_group( as_dict=True ) primary_key = Electrode.primary_key + if isinstance(electrode_list, ndarray): + # convert to list if it is an numpy array + electrode_list = list(electrode_list.astype(int).reshape(-1)) for e in all_electrodes: # create a dictionary so we can insert the electrodes if e["electrode_id"] in electrode_list: