line problems

Author: susanbao

src/maxdiffusion/tests/wan_transformer_test.py

@@ -310,7 +310,7 @@ def test_get_qt_provider(self, mock_qt_rule):
       weight_qtype=jnp.float8_e4m3fn,
       act_qtype=jnp.float8_e4m3fn
     )
-    
+
     # Case 4: Quantization enabled, type 'fp8_full'
     mock_qt_rule.reset_mock()
     config_fp8_full = Mock(spec=HyperParameters)
@@ -354,7 +354,7 @@ def test_quantize_transformer_enabled(self, mock_get_dummy_inputs, mock_quantize
     mock_pipeline = Mock()
     mock_mesh = Mock()
     mock_mesh.__enter__ = Mock(return_value=None)
-    mock_mesh.__exit__ = Mock(return_value=None) 
+    mock_mesh.__exit__ = Mock(return_value=None)
 
     # Mock the return values of dependencies
     mock_get_dummy_inputs.return_value = (Mock(), Mock(), Mock())