Update element_array_ephys/export/nwb/nwb.py

bendichter · CBroz1 · web-flow · commit 9e72773d7015 · 2022-03-21T16:55:41.000-06:00
Co-authored-by: Chris Brozdowski &lt;CBrozdowski@yahoo.com&gt;
diff --git a/element_array_ephys/export/nwb/nwb.py b/element_array_ephys/export/nwb/nwb.py
@@ -206,7 +206,7 @@ def create_units_table(
     )
 
     for unit in tqdm(
-        (clustering_query @ ephys.CuratedClustering.Unit).fetch(as_dict=True),
+        (ephys.CuratedClustering.Unit & clustering_query).fetch(as_dict=True),
         desc=f"creating units table for paramset {paramset_record['paramset_idx']}",
     ):
 

Original file line number	Diff line number	Diff line change
`@@ -206,7 +206,7 @@ def create_units_table(`
`206`	`206`	`)`
`207`	`207`
`208`	`208`	`for unit in tqdm(`
`209`		`- (clustering_query @ ephys.CuratedClustering.Unit).fetch(as_dict=True),`
	`209`	`+ (ephys.CuratedClustering.Unit & clustering_query).fetch(as_dict=True),`
`210`	`210`	`desc=f"creating units table for paramset {paramset_record['paramset_idx']}",`
`211`	`211`	`):`
`212`	`212`