def test_delete_tag():
"""
Confirm that fluent API delete tags actually works
:return:
"""
mlflow.set_tag("a", "b")
run = MlflowClient().get_run(mlflow.active_run().info.run_id)
assert "a" in run.data.tags
mlflow.delete_tag("a")
run = MlflowClient().get_run(mlflow.active_run().info.run_id)
assert "a" not in run.data.tags
with pytest.raises(MlflowException, match="No tag with name"):
mlflow.delete_tag("a")
with pytest.raises(MlflowException, match="No tag with name"):
mlflow.delete_tag("b")
mlflow.end_run()
def test_delete_tag():
"""
Confirm that fluent API delete tags actually works
:return:
"""
mlflow.set_tag('a', 'b')
run = MlflowClient().get_run(mlflow.active_run().info.run_id)
print(run.info.run_id)
assert 'a' in run.data.tags
mlflow.delete_tag('a')
run = MlflowClient().get_run(mlflow.active_run().info.run_id)
assert 'a' not in run.data.tags
with pytest.raises(MlflowException):
mlflow.delete_tag('a')
with pytest.raises(MlflowException):
mlflow.delete_tag('b')
mlflow.end_run()
#
# Code snippet for https://mlflow.org/docs/latest/python_api/mlflow.html#mlflow.delete_tag
#
import warnings
import mlflow
if __name__ == "__main__":
warnings.filterwarnings("ignore")
print(mlflow.__version__)
tags = {"engineering": "ML Platform", "engineering_remote": "ML Platform"}
with mlflow.start_run() as run:
mlflow.set_tags(tags)
with mlflow.start_run(run_id=run.info.run_id):
mlflow.delete_tag("engineering_remote")
