def set_experiment_tag(self, experiment_id: str,
tag: ExperimentTag) -> None:
_validate_experiment_tag(tag.key, tag.value)
experiment = self._get_experiment(experiment_id)
self._check_experiment_is_active(experiment)
new_tag = ElasticExperimentTag(key=tag.key, value=tag.value)
experiment.tags.append(new_tag)
experiment.update(tags=experiment.tags)
def set_experiment_tag(self, experiment_id, tag):
"""
Set a tag for the specified experiment
:param experiment_id: String ID of the experiment
:param tag: ExperimentRunTag instance to log
"""
_validate_experiment_tag(tag.key, tag.value)
with self.ManagedSessionMaker() as session:
experiment = self._get_experiment(session, experiment_id,
ViewType.ALL).to_mlflow_entity()
self._check_experiment_is_active(experiment)
session.merge(
SqlExperimentTag(experiment_id=experiment_id,
key=tag.key,
value=tag.value))
