Pickling DaskGeoDataFrame loses spatial_partitions

[TomAugspurger]

This is a bit of a strange use-case, but roundtripping a GeoDataFrame through pickle loses the spatial_partitions property:

import dask_geopandas
import geopandas.datasets
import pickle


df = geopandas.read_file(geopandas.datasets.get_path('naturalearth_lowres'))
ddf = dask_geopandas.from_geopandas(df, npartitions=2)
print(ddf.spatial_partitions)  # None
ddf2 = pickle.loads(pickle.dumps(df))
print(ddf2.spatial_partitions)  # AttributeError

__init__ doesn't run, so it's never set at

dask-geopandas/dask_geopandas/core.py

Line 67 in bc3af63

self.spatial_partitions = spatial_partitions

.

Maybe adding spatial partitions to _args to mirror dask.dataframe._Frame like in https://github.com/dask/dask/blob/main/dask/dataframe/core.py#L233-L241 would do the trick.