From 48d48a37acd37ddbca5330274fb402783188a825 Mon Sep 17 00:00:00 2001 From: Matthijs Douze Date: Wed, 17 May 2023 03:17:48 -0700 Subject: [PATCH] fix windows test (#2862) Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/2862 Fix windows test introduced by range search diff Reviewed By: algoriddle Differential Revision: D45901726 fbshipit-source-id: 16259b7718f1409adef814ea4c2b5707304849ca --- tests/test_build_blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_build_blocks.py b/tests/test_build_blocks.py index 189c78599..77ca92623 100644 --- a/tests/test_build_blocks.py +++ b/tests/test_build_blocks.py @@ -701,7 +701,7 @@ class TestMapInt64ToInt64(unittest.TestCase): """ test that we are able to lookup """ rs = np.random.RandomState(123) # make sure we have unique values - keys = np.unique(rs.choice(2 ** 29, size=n)) + keys = np.unique(rs.choice(2 ** 29, size=n).astype("int64")) rs.shuffle(keys) n = keys.size vals = rs.choice(2 ** 30, size=n).astype('int64')