Skip to content

Commit 4694a09

Browse files
Niharika KarnikNiharika Karnik
authored andcommitted
Changes to _constraints to fix the isin errors
1 parent 5b85295 commit 4694a09

File tree

1 file changed

+4
-12
lines changed

1 file changed

+4
-12
lines changed

pysensors/utils/_constraints.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)