Fix test

AnesBenmerzoug · Oct 14, 2024 · b62efdf · b62efdf
1 parent 454a213
commit b62efdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_weight.py b/tests/test_weight.py
@@ -11,7 +11,7 @@
     input_scores=numpy_st.arrays(
         dtype=st.one_of(st.just(np.float32), st.just(np.float64)),
         shape=numpy_st.array_shapes(min_dims=2, max_dims=2, min_side=1, max_side=5),
-        fill=st.floats(allow_nan=False, allow_infinity=False),
+        fill=st.floats(allow_nan=False, allow_infinity=False, width=32),
     ),
 )
 def test_identity_weights(input_scores: NDArray):