search

mlflow / mlflow-export-import

Apache License 2.0
132 stars 78 forks source link

Generic export model #65

Open danilopeixoto opened 1 year ago

danilopeixoto commented 1 year ago

Make generic import/export registered models.

danilopeixoto commented 1 year ago

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.