Skip to content

Commit d99a6c6

Browse files
committed
fix: fix random data type in tests
1 parent ee7045b commit d99a6c6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_custom_models.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -71,8 +71,8 @@ def test_mock_add_custom_models():
7171
source = ModelSource(hf="artificial")
7272

7373
num_tokens = 10
74-
dummy_pooled_embedding = np.random.random((1, dim))
75-
dummy_token_embedding = np.random.random((1, num_tokens, dim))
74+
dummy_pooled_embedding = np.random.random((1, dim)).astype(np.float32)
75+
dummy_token_embedding = np.random.random((1, num_tokens, dim)).astype(np.float32)
7676
dummy_attention_mask = np.ones((1, num_tokens)).astype(np.int64)
7777

7878
dummy_token_output = OnnxOutputContext(

0 commit comments

Comments
 (0)