Better exception message

secondmind-labs · Sep 10, 2024 · 69448cf · 69448cf
1 parent b95af67
commit 69448cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trieste/space.py b/trieste/space.py
@@ -656,8 +656,8 @@ def one_hot_encoder(self) -> EncoderFunction:
 
         def binary_encoder(x: TensorType) -> TensorType:
             # no need to one-hot encode binary categories (but we should still validate)
-            if not tf.reduce_all((x == 0) | (x == 1)):
-                raise ValueError(f"Invalid value {x}")
+            if tf.reduce_any((x != 0) & (x != 1)):
+                raise ValueError(f"Invalid values {tf.boolean_mask(x, ((x != 0) & (x != 1)))}")
             return x
 
         def encoder(x: TensorType) -> TensorType: