From b62efdf2ebc155477c307f6fd07149fdb5f314e0 Mon Sep 17 00:00:00 2001
From: Anes Benmerzoug <anes.benmerzoug@gmail.com>
Date: Mon, 14 Oct 2024 22:06:45 +0200
Subject: [PATCH] Fix test

---
 tests/test_weight.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_weight.py b/tests/test_weight.py
index 8f942a8..4b41100 100644
--- 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):