diff --git a/main.py b/main.py index 60dd47c45202e3f999b0424f500d4ea8e60c2902..08a5b06b3b787ae64c3bb94cf81d2b0aba64b5cf 100644 --- a/main.py +++ b/main.py @@ -272,21 +272,6 @@ def run_episode(evo_alg_str: str, debug: bool = False): if __name__ == "__main__": options = argument_parser.parse_args() - # log cli arguments - logger.info( - "CLI arguments:\n\tPositional:%s\n\tKeyword:\n\t\t%s", - ", ".join(options._get_args()), - "\n\t\t".join((f"{param}={value}" for param, value in options._get_kwargs())), - ) - - # debug mode will allow for a quick run - debug = options.debug - if debug is None: - debug = conv2bool(os.getenv("EP_DEBUG", False)) - if debug is None: - raise ValueError( - f"{os.getenv('EP_DEBUG')} is not allowed for env variable EP_DEBUG." - ) # set up evolution model match options.evolution_engine: @@ -314,6 +299,22 @@ if __name__ == "__main__": run_directory = initialize_run_directory(evolution_model) + # log cli arguments + logger.info( + "CLI arguments:\n\tPositional:%s\n\tKeyword:\n\t\t%s", + ", ".join(options._get_args()), + "\n\t\t".join((f"{param}={value}" for param, value in options._get_kwargs())), + ) + + # debug mode will allow for a quick run + debug = options.debug + if debug is None: + debug = conv2bool(os.getenv("EP_DEBUG", False)) + if debug is None: + raise ValueError( + f"{os.getenv('EP_DEBUG')} is not allowed for env variable EP_DEBUG." + ) + match options.task: case "sa": logger.info("Running with task sentiment analysis on dataset SetFit/sst2") diff --git a/utils.py b/utils.py index a88d47f947c433deff160a707371b451e91c3d41..128aa84c05d6d83429083b69f03306eb4239a5ae 100644 --- a/utils.py +++ b/utils.py @@ -29,7 +29,7 @@ def initialize_run_directory(model: OpenAI | Llama2): run_name = uuid4().hex else: run_name = run_name_match.group(0) - run_directory = current_directory / f"runs/run-{run_name}" + run_directory = current_directory / f"runs/{run_name}" run_directory.mkdir(parents=True, exist_ok=False) file_handler = logging.FileHandler(run_directory / "output.log") file_handler.setLevel(logging.DEBUG) @@ -37,7 +37,7 @@ def initialize_run_directory(model: OpenAI | Llama2): file_handler.setFormatter(formatter) logger.addHandler(file_handler) - logger.info(f"initialized run directory at {run_directory}") + logger.info(f"Hello my name is {run_name} and I live in {run_directory}") return run_directory