fix a bug in clean_knob_data()

This commit is contained in:
yangdsh
2020-04-13 05:43:09 +00:00
committed by Dana Van Aken
parent 9fe6ded36c
commit 147ef58065

View File

@@ -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)