diff --git a/chromadb/test/ef/test_multimodal_ef.py b/chromadb/test/ef/test_multimodal_ef.py index 52213c77a4c..82f66fea33e 100644 --- a/chromadb/test/ef/test_multimodal_ef.py +++ b/chromadb/test/ef/test_multimodal_ef.py @@ -55,6 +55,10 @@ def test_multimodal( n_examples: int = 10, n_query_results: int = 3, ) -> None: + # Fix numpy's random seed for reproducibility + random_state = np.random.get_state() + np.random.seed(0) + image_ids = [str(i) for i in range(n_examples)] images = [random_image() for _ in range(n_examples)] image_embeddings = default_ef(images) @@ -131,6 +135,7 @@ def test_multimodal( ) assert query_result["ids"][0] == nearest_document_neighbor_ids + np.random.set_state(random_state) @pytest.mark.xfail