A lot of stuff working :D. ResNet implemented!

1 files changed, 3 insertions, 3 deletions

diff --git a/src/training/prepare_experiments.py b/src/training/prepare_experiments.py
index 97c0304..4c3f9ba 100644
--- a/src/training/prepare_experiments.py
+++ b/src/training/prepare_experiments.py
@@ -7,11 +7,11 @@ from loguru import logger
 import yaml
 
 
-# flake8: noqa: S404,S607,S603
 def run_experiments(experiments_filename: str) -> None:
     """Run experiment from file."""
     with open(experiments_filename) as f:
         experiments_config = yaml.safe_load(f)
+
     num_experiments = len(experiments_config["experiments"])
     for index in range(num_experiments):
         experiment_config = experiments_config["experiments"][index]
@@ -27,10 +27,10 @@ def run_experiments(experiments_filename: str) -> None:
     type=str,
     help="Filename of Yaml file of experiments to run.",
 )
-def main(experiments_filename: str) -> None:
+def run_cli(experiments_filename: str) -> None:
     """Parse command-line arguments and run experiments from provided file."""
     run_experiments(experiments_filename)
 
 
 if __name__ == "__main__":
-    main()
+    run_cli()