TrueFoundry ML Client

from truefoundry.ml import get_client

client = get_client()

# Get all ML repos accessible by the current user
ml_repos = client.list_ml_repos()
print(f'Found {len(ml_repos)} ML repos:')
for repo in ml_repos:
    print(f'  - {repo}')

from truefoundry.ml import get_client

client = get_client()

# Create a new ML repository
client.create_ml_repo(
    name="my-ml-repo",
    storage_integration_fqn="your-storage-integration-fqn",
    description="My ML repository for experiments"
)

from truefoundry.ml import get_client

client = get_client()

client.create_run(
    ml_repo="my-ml-repo",
    run_name="my-run",
    tags={"key": "value"},
)

from truefoundry.ml import get_client

client = get_client()

client.get_run_by_id(
    run_id="run-id-123",
)

from truefoundry.ml import get_client

client = get_client()

client.get_run_by_fqn(
    run_fqn="value",
)

from truefoundry.ml import get_client

client = get_client()

client.get_run_by_name(
    ml_repo="my-ml-repo",
    run_name="my-run",
)

from truefoundry.ml import get_client

client = get_client()

client.search_runs(
    ml_repo="my-ml-repo",
    filter_string="value",
    run_view_type="value",
    order_by="value",
    job_run_name="value",
)

# Iterate through results
for item in response:
    print(item.name)

from truefoundry.ml import get_client

client = get_client()

client.get_model_version(
    ml_repo="my-ml-repo",
    name="my-run",
    version="value",
)

from truefoundry.ml import get_client

client = get_client()

client.get_model_version_by_fqn(
    fqn="run-id-123",
)

from truefoundry.ml import get_client

client = get_client()

client.list_model_versions(
    ml_repo="my-ml-repo",
    name="my-run",
)

from truefoundry.ml import get_client

client = get_client()

client.list_model_versions_by_fqn(
    model_fqn="value",
)

from truefoundry.ml import get_client

client = get_client()

client.get_artifact_version(
    ml_repo="my-ml-repo",
    name="my-run",
    artifact_type="value",
    version="value",
)

from truefoundry.ml import get_client

client = get_client()

client.get_artifact_version_by_fqn(
    fqn="run-id-123",
)

from truefoundry.ml import get_client

client = get_client()

client.list_artifact_versions(
    ml_repo="my-ml-repo",
    name="my-run",
    artifact_type="value",
)

from truefoundry.ml import get_client

client = get_client()

client.list_artifact_versions_by_fqn(
    artifact_fqn="value",
)

from truefoundry.ml import get_client

client = get_client()

client.log_artifact(
    ml_repo="my-ml-repo",
    name="my-run",
    artifact_paths=[ArtifactPath("file.txt")],
    description="value",
    metadata="value",
    progress="value",
)

from truefoundry.ml import get_client

client = get_client()

client.log_model(
    ml_repo="my-ml-repo",
    name="my-run",
    model_file_or_folder="model.pkl",
    description="value",
    metadata="value",
    progress="value",
    framework="value",
    environment="value",
)

from truefoundry.ml import get_client

client = get_client()

client.create_data_directory(
    ml_repo="my-ml-repo",
    name="my-run",
    description="value",
    metadata="value",
)

from truefoundry.ml import get_client

client = get_client()

client.get_data_directory_by_fqn(
    fqn="run-id-123",
)

from truefoundry.ml import get_client

client = get_client()

client.get_data_directory(
    ml_repo="my-ml-repo",
    name="my-run",
)

from truefoundry.ml import get_client

client = get_client()

client.list_data_directories(
    ml_repo="my-ml-repo",
)