From b7b0f7a4248e175993541b9eba1d0720be5cebf8 Mon Sep 17 00:00:00 2001 From: Future-Outlier Date: Tue, 24 Sep 2024 15:40:57 +0800 Subject: [PATCH] Update Databricks agent example to V2 (#1733) * use V2 instead Signed-off-by: Future-Outlier * Update examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py Co-authored-by: Kevin Su Signed-off-by: Future-Outlier * Update examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py Co-authored-by: Kevin Su Signed-off-by: Future-Outlier --------- Signed-off-by: Future-Outlier Co-authored-by: Kevin Su --- .../databricks_agent/databricks_agent_example_usage.py | 2 +- 1 file changed, 1 insertion(+), 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 index 9de652bc3..1fe9d3ff9 100644 --- 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]