Impl. CumlArray.deserialize()

The "constructor-kwargs" is removed in rapidsai/cudf#12009
madsbk · Nov 1, 2022 · 256a860 · 256a860
1 parent 3c147bf
commit 256a860
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/python/cuml/common/array.py b/python/cuml/common/array.py
@@ -349,6 +349,15 @@ def serialize(self) -> Tuple[dict, list]:
         frames = [CumlArray(f) for f in frames]
         return header, frames
 
+    @classmethod
+    @nvtx.annotate(message="common.CumlArray.deserialize", category="utils",
+                   domain="cuml_python")
+    def deserialize(cls, header, frames):
+        assert (
+            header["frame_count"] == 1
+        ), "Only expecting to deserialize Buffer with a single frame."
+        return cls(frames[0], **header["constructor-kwargs"])
+
     @nvtx.annotate(message="common.CumlArray.copy", category="utils",
                    domain="cuml_python")
     def copy(self) -> Buffer: