Skip to content

Provide interface for adding other metadata besides "metrics" #201

@jsnb-devoted

Description

@jsnb-devoted

I'm currently working on a migration from MLFlow and a model that we use logs parameters via the log_param method. It seems like I have to resort to logging the params as a artifact file because the SDK seems to only support logging "metrics" as metadata:

class ModelVersionMetadataSchema(TypedDict):
    metrics: NotRequired[dict[str, Any]]

Is there any plans to allow for arbitrary schemas of information in the metadata section? The params I'm looking to log are a simple dictionary. Params being a built in concept like MLFlow has would be nice but an all purpose interface would be fine too.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions