Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Make notebook-native auth work with more configurations of the Databricks Runtime #285

Merged
merged 5 commits into from
Aug 17, 2023
Merged

Make notebook-native auth work with more configurations of the Databricks Runtime #285

merged 5 commits into from
Aug 17, 2023

Commits on Aug 14, 2023

  1. Make notebook-native auth work with more configurations of the Databr… 

    …icks Runtime

This PR adds additional logging and hardening to `auth_type='runtime'`, which performs credentials lookup in the following order:

1. `init_runtime_native_auth` for the newest DBR versions.
2. `init_runtime_repl_auth` via Databricks REPL context and `workspaceUrl`.
3. `init_runtime_legacy_auth` via IPython context for legacy runtimes and modes.

Every detection step adds more logging on `DEBUG` level.

Based on publicly-accessible code in https://github.com/mlflow/mlflow/blame/6bd97bde24d78bcfbf6d50c1dd0f4fac2ed6987b/mlflow/utils/databricks_utils.py
    @nfx
    nfx committed Aug 14, 2023
    Configuration menu
    Copy the full SHA
    66d9ca0 View commit details
    Browse the repository at this point in the history

  2. fix tests

    @nfx
    nfx committed Aug 14, 2023
    Configuration menu
    Copy the full SHA
    bc6cb45 View commit details
    Browse the repository at this point in the history

  3. quick path to exit

    @nfx
    nfx committed Aug 14, 2023
    Configuration menu
    Copy the full SHA
    5f1b201 View commit details
    Browse the repository at this point in the history

Commits on Aug 17, 2023

  1. wip

    @nfx
    nfx committed Aug 17, 2023
    Configuration menu
    Copy the full SHA
    f94ca5b View commit details
    Browse the repository at this point in the history

  2. Add integration tests

    @nfx
    nfx committed Aug 17, 2023
    Configuration menu
    Copy the full SHA
    7a122fc View commit details
    Browse the repository at this point in the history