Update source/platforms/databricks.md

Co-authored-by: Jacob Tomlinson <[email protected]>
rapidsai · Nov 13, 2023 · fa497fe · fa497fe
1 parent e1b07b0
commit fa497fe
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/source/platforms/databricks.md b/source/platforms/databricks.md
@@ -92,10 +92,9 @@ Connect to the dask client and submit tasks.
 
 ```python
 from dask.distributed import Client
-from dask_databricks import DatabricksCluster
+import dask_databricks
 
-cluster = DatabricksCluster()
-client = Client(cluster)
+client = dask_databricks.get_client()
 
 def inc(x):
     return x + 1