def __call__()

in main_distributed.py [0:0]

• 22 lines of code
• 6 McCabe index (conditional complexity)

    def __call__(self):
        sel = self.sel
        fname = self.fname
        load_model = self.load_model
        logger.info(f'called-params {sel} {fname} {load_model}')

        # -- load script params
        params = None
        with open(fname, 'r') as y_file:
            params = yaml.load(y_file, Loader=yaml.FullLoader)
            if load_model is not None:
                params['meta']['load_checkpoint'] = load_model
            logger.info('loaded params...')
            pp = pprint.PrettyPrinter(indent=4)
            pp.pprint(params)

        logger.info('Running %s' % sel)
        if sel == 'paws_train':
            return paws(params)
        elif sel == 'suncet_train':
            return suncet(params)
        elif sel == 'fine_tune':
            return fine_tune(params)
        elif sel == 'snn_fine_tune':
            return snn_fine_tune(params)