def METHOD_NAME(self, enabled=True):
    self.learning_enabled[0] = int(enabled)  # type: ignore[operator]
    self.scale.requires_grad = enabled
    self.zero_point.requires_grad = enabled
    return self