src/transformers/integrations/integration_utils.py/CodeCarbonCallback/on_train_end
class CodeCarbonCallback:    def on_train_end(self, args, state, control, **kwargs):
        if self.tracker and state.is_local_process_zero:
            self.tracker.stop()