Fix numpy _whichmodule (#547)

cloudpipe · Jan 14, 2025 · caf55df · caf55df
1 parent 6220b0c
commit caf55df
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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):