fix a bug in clean_knob_data()
This commit is contained in:
parent
9fe6ded36c
commit
147ef58065
|
@ -127,7 +127,7 @@ def clean_knob_data(knob_matrix, knob_labels, session):
|
|||
new_lab = knob_name
|
||||
else:
|
||||
index = knob_labels.index(knob_name)
|
||||
new_col = knob_matrix[:, index]
|
||||
new_col = knob_matrix[:, index].reshape(-1, 1)
|
||||
new_lab = knob_labels[index]
|
||||
|
||||
new_labels.append(new_lab)
|
||||
|
|
Loading…
Reference in New Issue