test for boolean elements (MERLIN)

aeon-toolkit · Feb 3, 2025 · 7a00c17 · 7a00c17
1 parent 495d75a
commit 7a00c17
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/aeon/testing/estimator_checking/_yield_estimator_checks.py b/aeon/testing/estimator_checking/_yield_estimator_checks.py
@@ -624,6 +624,8 @@ def _equal_outputs(output1, output2):
         return False
     if np.issubdtype(type(output1), np.floating):
         return np.isclose(output1, output2)
+    if np.issubdtype(type(output1), np.bool_):
+        return output1 == output2
     if isinstance(output1, np.ndarray):  # 1. X an equal length collection or series
         if np.isscalar(output1):
             return np.allclose(output1, output2, equal_nan=True)