Update Databricks agent example to V2 (#1733)

* use V2 instead Signed-off-by: Future-Outlier <[email protected]> * Update examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py Co-authored-by: Kevin Su <[email protected]> Signed-off-by: Future-Outlier <[email protected]> * Update examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py Co-authored-by: Kevin Su <[email protected]> Signed-off-by: Future-Outlier <[email protected]> --------- Signed-off-by: Future-Outlier <[email protected]> Co-authored-by: Kevin Su <[email protected]>
flyteorg · Sep 24, 2024 · b7b0f7a · b7b0f7a
1 parent a94aa97
commit b7b0f7a
Showing 1 changed file with 1 addition and 1 deletion.
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 as Databricks
 
 
 # %% [markdown]