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