bioflow / open_biomed /utils /distributed.py
Rami-Troudi's picture
Phase 1: FastAPI integration with DeepPurpose DTI predictor
adecc9b
raw
history blame
415 Bytes
def setup_outputs_for_distributed(is_master, cfg):
"""
This function disables printing when not in master process
"""
import builtins as __builtin__
builtin_print = __builtin__.print
def print(*args, **kwargs):
force = kwargs.pop('force', False)
if is_master or force:
builtin_print(*args, **kwargs)
__builtin__.print = print
cfg.logging_level = "error"