Method _should_log

parallel/logging.py:13–19

Check if log should be performed.

(main_process_only)

    @staticmethod
    def _should_log(main_process_only):
        """Check if log should be performed."""
        if not dist.is_available() or not dist.is_initialized():
            is_main_process = True
        else:
            is_main_process = dist.get_rank() == 0
        return not main_process_only or (main_process_only and is_main_process)

    def log(self, level, msg, *args, **kwargs):
        """Delegates logger call after checking if we should log.