search

mlflow / mlflow-export-import

Apache License 2.0
110 stars 70 forks source link

Fix for model version import when its run is deleted #86

Closed amesar closed 1 year ago

amesar commented 1 year ago

Currently the import bombs on importing model versions if the version run has been deleted.

Account for deleted runs in migration with reference to model version imports

If a run is deleted in the source tracking server (lifecycle_stage=deleted), should we export it or not? Have an option, or just make an opinionated decisions?

amesar commented 1 year ago

Fix will not import the version if its run is not available and will display following error message and not stop overall import.

ERROR: Cannot import model version {'model': 'sklearn_wine', 'version': '2', 'stage': 'Production', 'run_id': '851b588a879441f7abc3dd0d295cef49'} since the source run_id was probably deleted.