code: def _compute_heating_rates(self):
          
          for varname, value in self.state.items():
              self.heating_rate[varname] = - self.b * (value - global_mean(value))
docs: Computes energy flux convergences to get heating rates in :math:`W/m^2`.