[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
openclimatefix · Oct 31, 2023 · ff5a91b · ff5a91b
1 parent 714ca3d
commit ff5a91b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -154,7 +154,7 @@ def get_data_config(
             )
 
         return data_config_file
-    
+
     def save_pretrained(
         self,
         save_directory: Union[str, Path],
@@ -199,7 +199,7 @@ def save_pretrained(
         if isinstance(config, dict):
             (save_directory / CONFIG_NAME).write_text(json.dumps(config, indent=4))
 <<<<<<< HEAD
-            
+
         # Save cleaned out datapipes configuration file
         make_clean_data_config(data_config, save_directory / DATA_CONFIG_NAME)
 =======

diff --git a/scripts/checkpoint_to_huggingface.py b/scripts/checkpoint_to_huggingface.py
@@ -58,8 +58,8 @@ def push_to_huggingface(
         checkpoint = torch.load(f"{checkpoint_dir_path}/last.ckpt")
 
     model.load_state_dict(state_dict=checkpoint["state_dict"])
-    
-    # Check for data config
+
+    # Check for data config
     data_config = f"{checkpoint_dir_path}/data_config.yaml"
     assert os.path.isfile(data_config)