new line added

Author: prishajain1

src/maxdiffusion/checkpointing/wan_checkpointer2_2.py

@@ -60,7 +60,7 @@ def load_wan_configs_from_orbax(self, step: Optional[int]) -> Tuple[Optional[dic
         return None, None
     max_logging.log(f"Loading WAN checkpoint from step {step}")
     metadatas = self.checkpoint_manager.item_metadata(step)
-    
+
     low_noise_transformer_metadata = metadatas.low_noise_transformer_state
     abstract_tree_structure_low_params = jax.tree_util.tree_map(ocp.utils.to_shape_dtype_struct, low_noise_transformer_metadata)
     low_params_restore = ocp.args.PyTreeRestore(
@@ -69,7 +69,7 @@ def load_wan_configs_from_orbax(self, step: Optional[int]) -> Tuple[Optional[dic
             abstract_tree_structure_low_params,
         )
     )
-    
+
     high_noise_transformer_metadata = metadatas.high_noise_transformer_state
     abstract_tree_structure_high_params = jax.tree_util.tree_map(ocp.utils.to_shape_dtype_struct, high_noise_transformer_metadata)
     high_params_restore = ocp.args.PyTreeRestore(

src/maxdiffusion/tests/wan_checkpointer2_2_test.py

@@ -110,4 +110,4 @@ def test_load_checkpoint_with_optimizer(self, mock_wan_pipeline, mock_create_man
 
 
 if __name__ == "__main__":
-  unittest.main()
+  unittest.main()