formatted

sleap_nn/config/trainer_config.py

@@ -98,7 +98,7 @@ def from_dict(cls, config_dict: Dict[Text, Any]) -> "TrainerJobConfig":
     @classmethod
     def from_json(cls, json_data: Text) -> "TrainingJobConfig":
         """Create TrainingJobConfig from JSON-formatted string
-        
+
         Arguments:
             json_data: JSON-formatted string that specifies the configurations.
 
@@ -137,7 +137,7 @@ def load_json(cls, filename: Text) -> "TrainingJobConfig":
             return cls.from_json(json_data)
 
     @classmethod
-    def load_yaml(cls, filename:Text) -> "TrainingJobConfig":
+    def load_yaml(cls, filename: Text) -> "TrainingJobConfig":
         """Load a training job configuration from a yaml file.
 
         Arguments:
@@ -155,7 +155,7 @@ def to_dict(self) -> DictConfig:
 
         Returns:
             The dictionary representation of the configuration.
-        """        
+        """
         return OmegaConf.structured(self)
 
     def to_json(self) -> str:
@@ -194,6 +194,7 @@ def save_yaml(self, filename: Text):
         with open(filename, "w") as f:
             f.write(self.to_yaml())
 
+
 @attrs.define
 class DataLoaderConfig:
     """train and val DataLoaderConfig: (Note: Any parameters from Torch's DataLoader could be used.)