ruffed

Author: grassesi

src/weathergen/train/trainer.py

@@ -26,7 +26,6 @@
 from weathergen.model.model import Model, ModelParams
 from weathergen.train.lr_scheduler import LearningRateScheduler
 from weathergen.train.trainer_base import Trainer_Base
-from weathergen.train.utils import get_run_id
 from weathergen.utils.config import Config
 from weathergen.utils.distributed import is_root
 from weathergen.utils.train_logger import TrainLogger
@@ -44,9 +43,12 @@ def __init__(self, checkpoint_freq=250, print_freq=10):
         self.print_freq = print_freq
 
     ###########################################
-    def init(self, cf: Config,):
+    def init(
+        self,
+        cf: Config,
+    ):
         self.cf = cf
-        
+
         assert cf.samples_per_epoch % cf.batch_size == 0
         assert cf.samples_per_validation % cf.batch_size_validation == 0
 

src/weathergen/utils/cli.py

@@ -116,5 +116,5 @@ def _add_model_loading_params(parser: argparse.ArgumentParser):
     parser.add_argument(
         "--reuse_run_id",
         action="store_true",
-        help="Use the id given via --from_run_id also for the current run."
+        help="Use the id given via --from_run_id also for the current run.",
     )

src/weathergen/utils/config.py

@@ -118,20 +118,20 @@ def load_config(
     """
     private_config = _load_private_conf(private_home)
     overwrite_configs = [_load_overwrite_conf(overwrite) for overwrite in overwrites]
-    
+
     if from_run_id is None:
         base_config = _load_default_conf()
     else:
         base_config = load_model_config(from_run_id, epoch, private_config["model_path"])
-    
 
     # use OmegaConf.unsafe_merge if too slow
     return OmegaConf.merge(base_config, private_config, *overwrite_configs)
 
+
 def set_run_id(config: Config, run_id: str | None, reuse_run_id: bool):
     """
     Determine run_id of current run.
-    
+
     Args:
         config: Base configuration loaded from previous run or default.
         run_id: Id assigned to this run. If None a new one will be generated.
@@ -142,9 +142,9 @@ def set_run_id(config: Config, run_id: str | None, reuse_run_id: bool):
             run_id = get_run_id()
 
         config.run_id = run_id
-    
+
     assert config.run_id is not None
-    
+
 
 def from_cli_arglist(arg_list: list[str]) -> Config:
     """