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

Add MLflow log_model option #1544

Open
wants to merge 21 commits into
base: main
Choose a base branch
from
Open

Add MLflow log_model option #1544

wants to merge 21 commits into from
+94 −45

Commits on Sep 24, 2024

  1. Register model with MLflow PySDK now that retries are baked in. This … 

    …cleans up the code a little and prevents us from having forked logic in Composer to fetch by run_id
    @nancyhung
    nancyhung committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    06d77db View commit details
    Browse the repository at this point in the history

  2. Register model with MLflow PySDK now that retries are baked in. This … 

    …cleans up the code a little and prevents us from having forked logic in Composer to fetch by run_id
    @nancyhung
    nancyhung committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    e40e5dd View commit details
    Browse the repository at this point in the history

  3. small changes

    @nancyhung
    nancyhung committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    454e18b View commit details
    Browse the repository at this point in the history

  4. isolated changes

    @nancyhung
    nancyhung committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    c8bd06f View commit details
    Browse the repository at this point in the history

Commits on Oct 2, 2024

  1. pr feedback with a print statement for testing

    @nancyhung
    nancyhung committed Oct 2, 2024
    Configuration menu
    Copy the full SHA
    0d3f9ce View commit details
    Browse the repository at this point in the history

Commits on Oct 4, 2024

  1. some more todos and need to test

    @nancyhung
    nancyhung committed Oct 4, 2024
    Configuration menu
    Copy the full SHA
    6ea8de5 View commit details
    Browse the repository at this point in the history

Commits on Oct 11, 2024

  1. need to test

    @nancyhung
    nancyhung committed Oct 11, 2024
    Configuration menu
    Copy the full SHA
    81306d8 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into nancy/log-model

    @nancyhung
    nancyhung committed Oct 11, 2024
    Configuration menu
    Copy the full SHA
    b854bb2 View commit details
    Browse the repository at this point in the history

Commits on Oct 15, 2024

  1. use mlflow log model by default

    @nancyhung
    nancyhung committed Oct 15, 2024
    Configuration menu
    Copy the full SHA
    bc73f65 View commit details
    Browse the repository at this point in the history

  2. patch push

    @nancyhung
    nancyhung committed Oct 15, 2024
    Configuration menu
    Copy the full SHA
    1915042 View commit details
    Browse the repository at this point in the history

Commits on Oct 22, 2024

  1. Merge branch 'main' into nancy/log-model

    @nancyhung
    nancyhung committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    bc29278 View commit details
    Browse the repository at this point in the history

  2. add log statements

    @nancyhung
    nancyhung committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    99589c7 View commit details
    Browse the repository at this point in the history

Commits on Oct 23, 2024

  1. add log outside of process

    @nancyhung
    nancyhung committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    04ddfaa View commit details
    Browse the repository at this point in the history

Commits on Oct 25, 2024

  1. fix

    @nancyhung
    nancyhung committed Oct 25, 2024
    Configuration menu
    Copy the full SHA
    8e42217 View commit details
    Browse the repository at this point in the history

  2. bug

    @nancyhung
    nancyhung committed Oct 25, 2024
    Configuration menu
    Copy the full SHA
    be04e3d View commit details
    Browse the repository at this point in the history

Commits on Oct 26, 2024

  1. print the registered model name

    @nancyhung
    nancyhung committed Oct 26, 2024
    Configuration menu
    Copy the full SHA
    5ab2cc7 View commit details
    Browse the repository at this point in the history

  2. update the model registry prefix

    @nancyhung
    nancyhung committed Oct 26, 2024
    Configuration menu
    Copy the full SHA
    79356d8 View commit details
    Browse the repository at this point in the history

  3. move the download code out of the if statement

    @nancyhung
    nancyhung committed Oct 26, 2024
    Configuration menu
    Copy the full SHA
    4327257 View commit details
    Browse the repository at this point in the history

  4. try registering just the model name

    @nancyhung
    nancyhung committed Oct 26, 2024
    Configuration menu
    Copy the full SHA
    6c5fb05 View commit details
    Browse the repository at this point in the history

  5. connect the existing mlflow run id

    @nancyhung
    nancyhung committed Oct 26, 2024
    Configuration menu
    Copy the full SHA
    bb0dd6a View commit details
    Browse the repository at this point in the history

  6. omg it works

    @nancyhung
    nancyhung committed Oct 26, 2024
    Configuration menu
    Copy the full SHA
    c5ae4ff View commit details
    Browse the repository at this point in the history