mosaicml / composer

Supercharge Your Model Training
http://docs.mosaicml.com
Apache License 2.0
5.12k stars 415 forks source link

Only requires `databricks-sdk` when inside the Databricks platform #3389

Closed antoinebrl closed 3 months ago

antoinebrl commented 3 months ago

What does this PR do?

We would like to use MLFlow outside of the databricks platform. For such scenario, the package databricks-sdk don't have to be installed. With this contribution the import of the databricks.sdk is gated by the presence of the DATABRICKS_TOKEN environment variable, making it possible to reduce the dependency chain.