Source code for mlflow.entities.model_registry.model_version_tag

from mlflow.entities.model_registry._model_registry_entity import _ModelRegistryEntity
from mlflow.protos.model_registry_pb2 import ModelVersionTag as ProtoModelVersionTag

[docs]class ModelVersionTag(_ModelRegistryEntity): """Tag object associated with a model version.""" def __init__(self, key, value): self._key = key self._value = value def __eq__(self, other): if type(other) is type(self): return self.__dict__ == other.__dict__ return False @property def key(self): """String name of the tag.""" return self._key @property def value(self): """String value of the tag.""" return self._value
[docs] @classmethod def from_proto(cls, proto): return cls(proto.key, proto.value)
[docs] def to_proto(self): tag = ProtoModelVersionTag() tag.key = self.key tag.value = self.value return tag