Mlflow не может получить модель, зарегистрированную в реестре моделей Gitlab ⇐ Python

[Anonymous]

Я пытаюсь загрузить артефакты и модель, которая была сохранена с использованием реестра моделей Gitlab.

Код: Выделить всё

# Running client
import os
from mlflow import MlflowClient

os.environ["MLFLOW_TRACKING_URI"] = "..."
os.environ["MLFLOW_TRACKING_TOKEN"] = "..."

client = MlflowClient()

# Model and version creation
model_name = 'test_model_n'

description = 'test model'
model = client.create_registered_model(model_name, description=description)

model_version = '1.0.1'
tags = { "gitlab.version": model_version }
model_version = client.create_model_version(model_name, model_version, tags=tags)

# Logging artifacts

run_id = model_version.run_id
client.log_artifact(run_id, 'mlartifacts/875771393960734527', artifact_path="")

< /code>
Теперь, чтобы принести модель, я попробовал несколько подходов: < /p>
 Метод 1 < /h4>
model_uri = f"models:/{model_name}/{model_version}"
model = mlflow.pyfunc.load_model(model_uri)

< /code>
Скачать полоса достичь 100%, но предупреждение и ошибка выскакивает: < /p>
WARNING:urllib3.connectionpool:Connection pool is full, discarding connection: "gitlab server" Connection pool size: 10
Could not find a registered artifact repository for: c:. Currently registered schemes are: ['', 'file', 's3', 'r2', 'gs', 'wasbs', 'ftp', 'sftp', 'dbfs', 'hdfs', 'viewfs', 'runs', 'models', 'http', 'https', 'mlflow-artifacts', 'abfss']
< /code>
 Метод 2 < /h4>
artifacts_path,  = mlflow.artifacts.download_artifacts(
run_id=run_id,
artifact_path="",
dst_path="./downloaded_model"

< /code>
ошибка: < /p>
ValueError: not enough values to unpack (expected 2, got 1)
< /code>
 Метод 3 < /h4>

# Se conosci il run_id specifico
mv = client.get_model_version(model_name, model_version)
run_id = mv.run_id

# Costruisci l'URI usando il run_id
run_uri = f"runs:/{run_id}/[path_to_the_model]"

model_from_run = mlflow.pyfunc.load_model(run_uri)
< /code>
ошибка: < /p>
ValueError: not enough values to unpack (expected 2, got 1)

Любая помощь будет очень оценена, не стесняйтесь просить дополнительную информацию, если вы думаете, что это поможет.

Подробнее здесь: https://stackoverflow.com/questions/797 ... l-registry