databricks-cli/internal/bundle/bundles/python_wheel_task/template/databricks.yml.tmpl

bundle:
  name: wheel-task

workspace:
  root_path: "~/.bundle/{{.unique_id}}"

{{if .python_wheel_wrapper}}
experimental:
  python_wheel_wrapper: true
{{end}}

resources:
  jobs:
    some_other_job:
      name: "[${bundle.target}] Test Wheel Job {{.unique_id}}"
      tasks:
        - task_key: TestTask
          new_cluster:
            num_workers: 1
            spark_version: "{{.spark_version}}"
            node_type_id: "{{.node_type_id}}"
            data_security_mode: USER_ISOLATION
            instance_pool_id: "{{.instance_pool_id}}"
          python_wheel_task:
            package_name: my_test_code
            entry_point: run
            parameters:
              - "one"
              - "two"
          libraries:
          - whl: ./dist/*.whl