def _run_init(argv): """Does one-time initialization and re-parses flags on rerun.""" if _run_init.done: return parse_flags_with_usage(argv) command_name.make_process_name_useful() # Set up absl logging handler. logging.use_absl_handler() argv = register_and_parse_flags_with_usage(argv=argv) if faulthandler: try: faulthandler.enable() except Exception: # pylint: disable=broad-except # Some tests verify stderr output very closely, so don't print anything. # Disabled faulthandler is a low-impact error. pass _run_init.done = True return argv
def _run_init(argv): """Does one-time initialization and re-parses flags on rerun.""" if _run_init.done: return parse_flags_with_usage(argv) command_name.make_process_name_useful() # Set up absl logging handler. logging.use_absl_handler() argv = _register_and_parse_flags_with_usage(argv=argv) if faulthandler: try: faulthandler.enable() except Exception: # pylint: disable=broad-except # Some tests verify stderr output very closely, so don't print anything. # Disabled faulthandler is a low-impact error. pass _run_init.done = True return argv
if ch == 'r': shutil.rmtree(str(save_dir)) elif ch == 'a': return save_dir.mkdir(exist_ok=True, parents=True) FLAGS.save_dir = save_dir logger.info(f'Saving log files to: {FLAGS.save_dir!s}') start = default_timer() expm.main(argv) dur = default_timer() - start logger.info(f'Experiment finished in {format_secs(dur)}') def initialize_logging(): logging.basicConfig(level=logging.INFO, format='[%(asctime)s] [%(name)s] [%(levelname)s] %(message)s') if __name__ == '__main__': command_name.make_process_name_useful() try: initialize_logging() sys.exit(main()) except UsageError as error: usage(shorthelp=True, detailed_error=error, exitcode=error.exitcode) finally: reap_children()
def test_make_process_name_useful(self): test_name = 'hello.from.test' with mock.patch('sys.argv', [test_name]): command_name.make_process_name_useful() self.assertProcessNameSimilarTo(test_name)