diff --git a/python_modules/libraries/dagster-databricks/dagster_databricks/databricks.py b/python_modules/libraries/dagster-databricks/dagster_databricks/databricks.py index 9cd64f125b2cf..a15bb4b3e1954 100644 --- a/python_modules/libraries/dagster-databricks/dagster_databricks/databricks.py +++ b/python_modules/libraries/dagster-databricks/dagster_databricks/databricks.py @@ -564,7 +564,7 @@ def submit_run(self, run_config: Mapping[str, Any], task: Mapping[str, Any]) -> new_cluster["instance_pool_id"] = nodes["instance_pool_id"] if "driver_instance_pool_id" in nodes: - new_cluster["driver_node_type_id"] = nodes["driver_node_type_id"] + new_cluster["driver_instance_pool_id"] = nodes["driver_instance_pool_id"] else: node_types = nodes["node_types"] new_cluster["node_type_id"] = node_types["node_type_id"]