def METHOD_NAME(cls, _root, info: ResolveInfo, /, **data):
    cls.validate_input(info, data)
    obj = cls.get_object(info, data["id"])
    obj.tax_exemption = data["tax_exemption"]
    if isinstance(obj, Checkout):
        cls._invalidate_checkout_prices(info, obj)
        obj.save(update_fields=["tax_exemption", "price_expiration", "last_change"])
    if isinstance(obj, Order):
        cls.validate_order_status(obj)
        obj.should_refresh_prices = True
        obj.save(
            update_fields=["tax_exemption", "should_refresh_prices", "updated_at"]
        )
    return TaxExemptionManage(taxable_object=obj)