Better logic for probe_planar_contour eq

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