mlflow-export-import
mlflow-export-import copied to clipboard
Generic export model
Make generic import/export registered models.
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.