Bugfix to behavior that created 2D clu_ids

Author: berkgercek

brainbox/modeling/neural_model.py

@@ -103,7 +103,7 @@ def __init__(self, design_matrix, spk_times, spk_clu,
         self.design = design_matrix
         self.spikes = spks
         self.clu = clu
-        self.clu_ids = np.argwhere(np.sum(trialspiking, axis=0) > mintrials)
+        self.clu_ids = np.argwhere(np.sum(trialspiking, axis=0) > mintrials).flatten()
         self.traininds = traininds
         self.testinds = testinds
         self.stepwise = stepwise