Turning on dask-expr for dask-sql to work.

Author: alxmrs

demo/sst.py

@@ -3,6 +3,7 @@
 
 Please run the following to set up cloud resources:
 ```
+pip install ".[demo]"
 gcloud auth application-default login
 coiled login
 coiled setup gcp --region us-central1
@@ -62,6 +63,7 @@ def rand_wx(times) -> xr.Dataset:
   )
 
 
+# TODO(alxmrs): Make spot instances a flag.
 parser = argparse.ArgumentParser()
 parser.add_argument('--timeframe', choices=TIMEFRAMES.keys(), default='day')
 parser.add_argument(

pyproject.toml

@@ -32,12 +32,17 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
+io = [
+    "xarray[io]",
+    "gcsfs",
+]
 test = [
-  "pytest",
-  "xarray[io]",
-  "gcsfs",
+    "xarray_sql[io]",
+    "pytest",
 ]
+
 demo = [
+    "xarray_sql[io]",
     "coiled"
 ]
 

xarray_sql/df.py

@@ -10,6 +10,12 @@
 Block = t.Dict[t.Hashable, slice]
 Chunks = t.Optional[t.Dict[str, int]]
 
+# Turn on Dask-Expr
+dask.config.set({'dataframe.query-planning-warning': False})
+dask.config.set({'dataframe.query-planning': True})
+# Turn on Copy-On-Write (needs Pandas 2.0).
+pd.options.mode.copy_on_write = True
+
 
 # Borrowed from Xarray
 def _get_chunk_slicer(