def METHOD_NAME(self, output):
    if output == 'gravity':
        out = self.model._interpolator.aesara_graph.aesara_output()[12][0]
    elif output == 'lith':
        out = self.model._interpolator.aesara_graph.aesara_output()[0][1]
    else:
        raise AttributeError()
    return out