File tree Expand file tree Collapse file tree 1 file changed +4
-9
lines changed Expand file tree Collapse file tree 1 file changed +4
-9
lines changed Original file line number Diff line number Diff line change @@ -1043,25 +1043,20 @@ def make(self, key):
1043
1043
) # {channel: electrode}
1044
1044
1045
1045
# Get unit id to quality label mapping
1046
- cluster_quality_label_map = {}
1047
1046
try :
1048
1047
cluster_quality_label_map = pd .read_csv (
1049
1048
sorting_dir / "sorter_output" / "cluster_KSLabel.tsv" ,
1050
1049
delimiter = "\t " ,
1051
1050
)
1051
+ except FileNotFoundError :
1052
+ cluster_quality_label_map = {}
1053
+ else :
1052
1054
cluster_quality_label_map : dict [
1053
1055
int , str
1054
1056
] = cluster_quality_label_map .set_index ("cluster_id" )[
1055
1057
"KSLabel"
1056
1058
].to_dict () # {unit: quality_label}
1057
- except FileNotFoundError :
1058
- pass
1059
-
1060
- # Get channel to electrode mapping
1061
- channel2depth_map = dict (
1062
- zip (* electrode_query .fetch ("channel_idx" , "y_coord" ))
1063
- ) # {channel: depth}
1064
-
1059
+
1065
1060
peak_electrode_ind = np .array (
1066
1061
[
1067
1062
channel_info [unit_peak_channel_map [unit_id ]]["electrode" ]
You can’t perform that action at this time.
0 commit comments