def _gather_saveables_for_checkpoint(self):
    if (not self._is_local):
        return {'tree_ensemble': self._saveable}
def _gather_saveables_for_checkpoint(self):
    if (not self._is_local):
        return {'tree_ensemble': self._saveable}