def METHOD_NAME(self):
    """Test that a custom gradient transform is properly labelled"""
    dev = qml.device("default.qubit", wires=2)
    @qml.gradients.gradient_transform
    def my_transform(tape):
        return tape, None
    @qml.qnode(dev, diff_method=my_transform)
    def circuit():
        return qml.probs(wires=0)
    info = qml.specs(circuit)()
    assert info["diff_method"] == "test_specs.my_transform"
    assert info["gradient_fn"] == "test_specs.my_transform"