Add planar contour in equality

SpikeInterface · Feb 5, 2024 · 43828a4 · 43828a4
1 parent dc7066e
commit 43828a4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/probeinterface/probe.py b/src/probeinterface/probe.py
@@ -515,6 +515,11 @@ def __eq__(self, other):
             if not np.array_equal(self.contact_annotations[key], other.contact_annotations[key]):
                 return False
 
+        # planar contour
+        if self.probe_planar_contour is not None:
+            if not np.array_equal(self.probe_planar_contour, other.probe_planar_contour):
+                return False
+
         return True
 
     def copy(self):