use V2 instead

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Sep 21, 2024 · d3876ee · d3876ee
1 parent a94aa97
commit d3876ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py b/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py
@@ -11,7 +11,7 @@
 
 import flytekit
 from flytekit import Resources, task, workflow
-from flytekitplugins.spark import Databricks
+from flytekitplugins.spark import DatabricksV2
 
 
 # %% [markdown]
@@ -20,7 +20,7 @@
 # [Databricks job request](https://docs.databricks.com/dev-tools/api/2.0/jobs.html#request-structure) documentation.
 # %%
 @task(
-    task_config=Databricks(
+    task_config=DatabricksV2(
         spark_conf={
             "spark.driver.memory": "1000M",
             "spark.executor.memory": "1000M",