Skip to content

5649 auto close mlflow run#5656

Merged
wyli merged 6 commits into
Project-MONAI:devfrom
binliunls:5649-auto-close-mlflow-run
Dec 6, 2022
Merged

5649 auto close mlflow run#5656
wyli merged 6 commits into
Project-MONAI:devfrom
binliunls:5649-auto-close-mlflow-run

Conversation

@binliunls

Copy link
Copy Markdown
Contributor

Fixes #5649 .

Description

Add the auto_close parameter to mlflow_handler so that it can close the mlflow run according to given value.
Update the default handler parameter, so that trainer and evaluator could have True value on auto_close.

Types of changes

  • Non-breaking change (fix or new feature that would not break existing functionality).
  • Breaking change (fix or new feature that would cause existing functionality to change).
  • New tests added to cover the changes.
  • Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
  • Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.
  • In-line docstrings updated.
  • Documentation updated, tested make html command in the docs/ folder.

Signed-off-by: binliu <binliu@nvidia.com>
…rue and modify the corresponding docstring

Signed-off-by: binliu <binliu@nvidia.com>

@Nic-Ma Nic-Ma left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me, please change the unit test.

Thanks.

…lose parameter in mlflow_handler can implement this function

Signed-off-by: binliu <binliu@nvidia.com>
@Nic-Ma

Nic-Ma commented Dec 5, 2022

Copy link
Copy Markdown
Contributor

/build

@Nic-Ma Nic-Ma changed the title [WIP] 5649 auto close mlflow run 5649 auto close mlflow run Dec 5, 2022
@Nic-Ma Nic-Ma marked this pull request as ready for review December 5, 2022 16:43
@Nic-Ma Nic-Ma requested a review from wyli December 5, 2022 16:43
Comment thread monai/handlers/mlflow_handler.py Outdated
@wyli

wyli commented Dec 6, 2022

Copy link
Copy Markdown
Contributor

/build

@wyli wyli enabled auto-merge (squash) December 6, 2022 07:38
@wyli wyli merged commit e457a5a into Project-MONAI:dev Dec 6, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Close MLFlow run automatically when workflow is end

3 participants