mlflow-export-import icon indicating copy to clipboard operation
mlflow-export-import copied to clipboard

Generic export model

Open danilopeixoto opened this issue 2 years ago • 1 comments

Make generic import/export registered models.

danilopeixoto avatar Dec 22 '22 14:12 danilopeixoto

Hi @amesar,

I believe the model importing is failing due thread safety when --use-threads true.

The model registry is resolved just for the first model or fails for the first model and just works for next models:

mlflow.tracking.registry.UnsupportedModelRegistryStoreURIException:  Model registry functionality is unavailable; got unsupported URI 'custom://profile' for model registry data storage. Supported URI schemes are: ['databricks', 'http', 'https', 'postgresql', 'mysql', 'sqlite', 'mssql', '', 'file']

When using --use-threads false everything works well, all models are imported.

I did comment set_experiment (global variable) without affecting results... But the import is still failing - maybe the model registry LRU cache is not working well with threads.

danilopeixoto avatar Dec 23 '22 15:12 danilopeixoto