Unity Catalog models export/Import not working

[WAG10]

Hi team,
We are working on UC models export and import from one workspace to another. Export seems somehpow happening but for import we get dbfs path error.
command for import : import-model --input-dir --model --experiment-name

Error :
404 Client Error: The specified blob does not exist. for url: https://dbstorageo2kgj4j4r3vkg.blob.core.windows.net/jobs/4431780668092041/mlflow-tracking/997485493456103/8f15851f15dc4ef2834e2aa58db2ede4/artifacts/3dade68ff6b542e68d583338b525578c/artifacts/quality_prediction?sig=WLEZ2LgUdJ%2FcTqrL3CxEKWrwfJ2opzY3ZXlbY7kafUY%3D&se=2024-06-07T12%3A01%3A05Z&sv=2019-02-02&spr=https&sp=r&sr=b. Response text: BlobNotFoundThe specified blob does not exist.
RequestId:067b193d-401e-006d-40c9-b86de0000000
Time:2024-06-07T11:01:05.8831272Z

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/usr/local/bin/import-model", line 11, in
load_entry_point('mlflow-export-import==1.2.0', 'console_scripts', 'import-model')()
File "/usr/lib/python3/dist-packages/click/core.py", line 764, in call
return self.main(*args, **kwargs)
File "/usr/lib/python3/dist-packages/click/core.py", line 717, in main
rv = self.invoke(ctx)
File "/usr/lib/python3/dist-packages/click/core.py", line 956, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/usr/lib/python3/dist-packages/click/core.py", line 555, in invoke
return callback(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/mlflow_export_import-1.2.0-py3.8.egg/mlflow_export_import/model/import_model.py", line 346, in main
import_model(
File "/usr/local/lib/python3.8/dist-packages/mlflow_export_import-1.2.0-py3.8.egg/mlflow_export_import/model/import_model.py", line 58, in import_model
return importer.import_model(
File "/usr/local/lib/python3.8/dist-packages/mlflow_export_import-1.2.0-py3.8.egg/mlflow_export_import/model/import_model.py", line 166, in import_model
self.import_version(model_name, vr, run_id)
File "/usr/local/lib/python3.8/dist-packages/mlflow_export_import-1.2.0-py3.8.egg/mlflow_export_import/model/import_model.py", line 215, in import_version
return _import_model_version(
File "/usr/local/lib/python3.8/dist-packages/mlflow_export_import-1.2.0-py3.8.egg/mlflow_export_import/model_version/import_model_version.py", line 119, in _import_model_version
dst_vr = mlflow_client.create_model_version(
File "/usr/local/lib/python3.8/dist-packages/mlflow/tracking/client.py", line 3687, in create_model_version
return self._create_model_version(
File "/usr/local/lib/python3.8/dist-packages/mlflow/tracking/client.py", line 3598, in _create_model_version
return self._get_registry_client().create_model_version(
File "/usr/local/lib/python3.8/dist-packages/mlflow/tracking/_model_registry/client.py", line 215, in create_model_version
mv = self.store.create_model_version(
File "/usr/local/lib/python3.8/dist-packages/mlflow/store/_unity_catalog/registry/rest_store.py", line 737, in create_model_version
with self._local_model_dir(source, local_model_path) as local_model_dir:
File "/usr/lib/python3.8/contextlib.py", line 113, in enter
return next(self.gen)
File "/usr/local/lib/python3.8/dist-packages/mlflow/store/_unity_catalog/registry/rest_store.py", line 672, in _local_model_dir
raise MlflowException(
mlflow.exceptions.MlflowException: Unable to download model artifacts from source artifact location 'dbfs:/databricks/mlflow-tracking/997485493456103/8f15851f15dc4ef2834e2aa58db2ede4/artifacts/3dade68ff6b542e68d583338b525578c/artifacts/quality_prediction' in order to upload them to Unity Catalog. Please ensure the source artifact location exists and that you can download from it via mlflow.artifacts.download_artifacts()