Merge pull request #595 from ikrommyd/handle-virtual-array-in-pickling

feat: handle virtual array in pickling

Author: pfackeldey

src/dask_awkward/pickle.py

@@ -15,6 +15,8 @@ def _maybe_make_pickle_buffer(buffer: Any) -> PlaceholderArray | PickleBuffer:
     if isinstance(buffer, PlaceholderArray):
         return buffer
     else:
+        if hasattr(buffer, "materialize") and callable(buffer.materialize):
+            buffer = buffer.materialize()
         return PickleBuffer(buffer)