Update maskit/masks.py

Co-authored-by: Max Fischer <maxfischer2781@gmail.com>
cirKITers · May 10, 2021 · ea6de45 · ea6de45
1 parent fbcfc27
commit ea6de45
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/maskit/masks.py b/maskit/masks.py
@@ -300,11 +300,11 @@ def mask_changed(self, mask: Mask, indices: np.ndarray):
             return
         np_indices = tuple(zip(*indices))
         if not np.all(mask.mask[np_indices]):
-            if self.wire_mask == mask:
+            if self.wire_mask is mask:
                 self.parameters[:, np_indices] = self.default_value
-            elif self.layer_mask == mask:
+            elif self.layer_mask is mask:
                 self.parameters[np_indices, :] = self.default_value
-            elif self.parameter_mask == mask:
+            elif self.parameter_mask is mask:
                 self.parameters[np_indices] = self.default_value
             else:
                 raise NotImplementedError(f"The mask {mask} is not supported")