How to get model references logged by a specific run?

[polyaxon-team]

We are trying to save a model using log_model_ref and add a name to it, i.e. best_auc. Then we want to be able to retrieve this model from the latest run.
However, if we use RunClient.client.runs_v1.get_runs_artifacts_lineage this returns all the artifacts ever generated for that project. And if we use RunClient.get_artifacts_tree, we do have more control about which run we are looking at, but we lose the name information we set when using log_model_ref?

[polyaxon-team]

To get the logged model refs:

from polyaxon.client import RunClient

run_client = RunClient(project="PROJECT_NAME", run_uuid="RUN_UUID")

# Query the lineage information
lineages = run_client.get_artifacts_lineage(query="kind: model").results

# Download the lineage assets
for lineage in lineages:
    run_client.download_artifact_for_lineage(lineage=lineage)

You can restrict the ref to specific lineage by filtering further by name:

lineages = run_client.get_artifacts_lineage(query="kind: model, name: best_auc").results