diff --git a/test/test_cpp_extensions_open_device_registration.py b/test/test_cpp_extensions_open_device_registration.py index ba90aad46fe..230cce990b8 100644 --- a/test/test_cpp_extensions_open_device_registration.py +++ b/test/test_cpp_extensions_open_device_registration.py @@ -534,6 +534,10 @@ class TestCppExtensionOpenRgistration(common.TestCase): np.__version__ < "1.25", "versions < 1.25 serialize dtypes differently from how it's serialized in data_legacy_numpy", ) + @unittest.skipIf( + np.__version__ >= "2.1", + "weights_only failure on numpy >= 2.1", + ) def test_open_device_numpy_serialization(self): """ This tests the legacy _rebuild_device_tensor_from_numpy serialization path