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))