Support for scipy 1.12 (#8474)

dask · Jan 22, 2024 · 6d4b409 · 6d4b409
1 parent 564149b
commit 6d4b409
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/distributed/protocol/scipy.py b/distributed/protocol/scipy.py
@@ -14,7 +14,7 @@
 register_generic(scipy.sparse.spmatrix, "dask", dask_serialize, dask_deserialize)
 
 
-@dask_serialize.register(scipy.sparse.dok.dok_matrix)
+@dask_serialize.register(scipy.sparse.dok_matrix)
 def serialize_scipy_sparse_dok(x):
     coo_header, coo_frames = dask_serialize(x.tocoo())
 
@@ -24,7 +24,7 @@ def serialize_scipy_sparse_dok(x):
     return header, frames
 
 
-@dask_deserialize.register(scipy.sparse.dok.dok_matrix)
+@dask_deserialize.register(scipy.sparse.dok_matrix)
 def deserialize_scipy_sparse_dok(header, frames):
     coo_header = header["coo_header"]
     coo_frames = frames