dbx by Databricks Labs

logo

DataBricks CLI eXtensions - aka dbx is a CLI tool for advanced Databricks jobs management.

Documentation Status Latest Python Release build codecov lgtm-alerts lgtm-code-quality Total Downloads We use black for formatting

Concept

dbx simplifies jobs launch and deployment process across multiple environments. It also helps to package your project and deliver it to your Databricks environment in a versioned fashion. Designed in a CLI-first manner, it is built to be actively used both inside CI/CD pipelines and as a part of local tooling for fast prototyping.

Requirements

  • Python Version > 3.6

  • pip or conda

Installation

  • with pip:

pip install dbx

Quickstart

Please refer to the Quickstart section.

Documentation

Please refer to the docs page.

Differences from other tools

Tool

Comment

databricks-cli

dbx is NOT a replacement for databricks-cli. Quite the opposite - dbx is heavily dependent on databricks-cli and uses most of the APIs exactly from databricks-cli SDK.

mlflow cli

dbx is NOT a replacement for mlflow cli. dbx uses some of the MLflow APIs under the hood to store serialized job objects, but doesn’t use mlflow CLI directly.

Databricks Terraform Provider

While dbx is primarily oriented on versioned job management, Databricks Terraform Provider provides much wider set of infrastructure settings. In comparison, dbx doesn’t provide infrastructure management capabilities, but brings more flexible deployment and launch options.

cicd-templates

cicd-templates is a Python project template, which actively uses dbx for jobs management and CI-related operations. You can choose, whenever you would like to use this template, or use dbx separately and choose the project structure on your own.

Databricks Stack CLI

Databricks Stack CLI is a great component for managing a stack of objects. dbx concentrates on the versioning and packaging jobs together, not treating files and notebooks as a separate component.

Limitations

  • Python > 3.6

  • dbx execute can only be used on clusters with Databricks ML Runtime 7.X and only for Python-based projects.

Versioning

For CLI interfaces, we support SemVer approach. However, for API components we don’t use SemVer as of now. This may lead to instability when using dbx API methods directly.

Feedback

Issues with dbx? Found a bug? Have a great idea for an addition? Feel free to file an issue.

Contributing

Please find more details about contributing to dbx in the contributing doc.