From caf55dfc9d6c933a743cac2020f10334ba263221 Mon Sep 17 00:00:00 2001
From: Olivier Grisel <olivier.grisel@ensta.org>
Date: Tue, 14 Jan 2025 14:15:54 +0100
Subject: [PATCH] Fix numpy _whichmodule (#547)

---
 tests/cloudpickle_test.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/cloudpickle_test.py b/tests/cloudpickle_test.py
index 1e594784..1489d3f0 100644
--- a/tests/cloudpickle_test.py
+++ b/tests/cloudpickle_test.py
@@ -1527,7 +1527,8 @@ def test_importing_multiprocessing_does_not_impact_whichmodule(self):
         self.assertEqual(proc.wait(), 0)
         assert out.strip() in (
             b"numpy.core._multiarray_umath",  # numpy 1
-            b"numpy._core._multiarray_umath",  # numpy 2
+            b"numpy._core._multiarray_umath",  # older numpy 2
+            b"numpy",  # more recent numpy 2
         )
 
     def test_unrelated_faulty_module(self):