From b431ae399def3a749833fe81c2c291b016cf3213 Mon Sep 17 00:00:00 2001 From: Future-Outlier Date: Fri, 27 Sep 2024 00:26:06 +0800 Subject: [PATCH] nit Signed-off-by: Future-Outlier --- .../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 f2e5c7424..1fe9d3ff9 100644 --- a/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py +++ b/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py @@ -20,7 +20,7 @@ # [Databricks job request](https://docs.databricks.com/dev-tools/api/2.0/jobs.html#request-structure) documentation. # %% @task( - task_config=DatabricksV2( + task_config=Databricks( spark_conf={ "spark.driver.memory": "1000M", "spark.executor.memory": "1000M",