Merge pull request #57 from martindurant/dask-fix

Fix dask tests

tests/test_dask.py

@@ -2,7 +2,7 @@
 import pyarrow as pa
 import pytest
 
-import awkward_pandas  # noqa
+import awkward_pandas.dask  # noqa
 
 dd = pytest.importorskip("dask.dataframe")
 
@@ -36,7 +36,9 @@ def test_accessor():
 def test_distributed():
     distributed = pytest.importorskip("distributed")
 
-    with distributed.Client(n_workers=1, threads_per_worker=1):
+    with distributed.Client(
+        n_workers=1, threads_per_worker=1, preload=["awkward_pandas.dask"]
+    ):
         data = pd.arrays.ArrowExtensionArray(pa.array([[0], [0, 1]] * 2))
         s = pd.Series(data)
         df = pd.DataFrame({"s": s})