test_to_categorical: Ensure that label_smoothing < 0.5 (explosion…

…#680) * `test_to_categorical`: Ensure that `label_smoothing < 0.5` * Use `exclude_max` instead of clamping to `0.49`
adrianeboyd · Jun 2, 2022 · b13840e · b13840e
1 parent 6ab4cc6
commit b13840e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/thinc/tests/test_util.py b/thinc/tests/test_util.py
@@ -46,7 +46,9 @@ def test_array_module_cpu_gpu_helpers():
     assert not is_numpy_array((1, 2))
 
 
-@given(label_smoothing=strategies.floats(min_value=0.0, max_value=0.5))
+@given(
+    label_smoothing=strategies.floats(min_value=0.0, max_value=0.5, exclude_max=True)
+)
 def test_to_categorical(label_smoothing):
     # Test without n_classes
     one_hot = to_categorical(numpy.asarray([1, 2], dtype="i"))