removed extra tests

Author: prishajain1

src/maxdiffusion/tests/wan_checkpointer_test.py

@@ -239,27 +239,6 @@ def test_load_checkpoint_with_optimizer_in_high_noise(self, mock_wan_pipeline, m
     self.assertEqual(opt_state["learning_rate"], 0.002)
     self.assertEqual(step, 1)
 
-  @patch("maxdiffusion.checkpointing.wan_checkpointer.create_orbax_checkpoint_manager")
-  def test_save_checkpoint(self, mock_create_manager):
-    """Test saving checkpoint for WAN 2.2."""
-    mock_manager = MagicMock()
-    mock_create_manager.return_value = mock_manager
-
-    mock_pipeline = MagicMock()
-    mock_pipeline.low_noise_transformer.to_json_string.return_value = '{"config": "test"}'
-
-    train_states = {
-        "low_noise_transformer": {"params": {}, "opt_state": {}},
-        "high_noise_transformer": {"params": {}}
-    }
-
-    checkpointer = WanCheckpointer2_2(model_key=self.config.model_key, config=self.config)
-    checkpointer.save_checkpoint(train_step=100, pipeline=mock_pipeline, train_states=train_states)
-
-    mock_manager.save.assert_called_once()
-    call_args = mock_manager.save.call_args
-    self.assertEqual(call_args[0][0], 100)  # train_step
-
 
 class WanCheckpointerFactoryTest(unittest.TestCase):
   """Tests for checkpointer factory/selection logic."""