@@ -555,12 +555,8 @@ def plot_selected_sensors(
555555 """
556556 n_samples , n_features = self .data .shape
557557 n_sensors = len (sensors )
558- constrained = sensors [
559- np .where (~ np .isin (all_sensors [:n_sensors ], sensors ))[0 ]
560- ]
561- unconstrained = sensors [
562- np .where (np .isin (all_sensors [:n_sensors ], sensors ))[0 ]
563- ]
558+ constrained = sensors [np .where (~ np .isin (all_sensors [:n_sensors ], sensors ))[0 ]]
559+ unconstrained = sensors [np .where (np .isin (all_sensors [:n_sensors ], sensors ))[0 ]]
564560 if isinstance (self .data , np .ndarray ):
565561 xconst = np .mod (constrained , np .sqrt (n_features ))
566562 yconst = np .floor (constrained / np .sqrt (n_features ))
@@ -643,12 +639,8 @@ def annotate_sensors(
643639 """
644640 n_samples , n_features = self .data .shape
645641 n_sensors = len (sensors )
646- constrained = sensors [
647- np .where (np .isin (all_sensors [:n_sensors ], sensors ) is False )[0 ]
648- ]
649- unconstrained = sensors [
650- np .where (np .isin (all_sensors [:n_sensors ], sensors ) is True )[0 ]
651- ]
642+ constrained = sensors [np .where (~ np .isin (all_sensors [:n_sensors ], sensors ))[0 ]]
643+ unconstrained = sensors [np .where (np .isin (all_sensors [:n_sensors ], sensors ))[0 ]]
652644 if isinstance (self .data , np .ndarray ):
653645 xTop = np .mod (sensors , np .sqrt (n_features ))
654646 yTop = np .floor (sensors / np .sqrt (n_features ))
0 commit comments