fix

Signed-off-by: 李子琦 <[email protected]>

Author: Liziqi-77

tests/ut/distributed/mooncake/test_config_data.py

@@ -9,25 +9,25 @@ class TestParseGlobalSegmentSize(unittest.TestCase):
     def test_int_input(self):
         self.assertEqual(_parse_global_segment_size(1024), 1024)
         self.assertEqual(_parse_global_segment_size(0), 0)
-    
+
     def test_gb_unit(self):
         self.assertEqual(_parse_global_segment_size("2GB"), 2 * 1024**3)
         self.assertEqual(_parse_global_segment_size("1.5GB"),
                          int(1.5 * 1024**3))
         self.assertEqual(_parse_global_segment_size(" 2 GB "), 2 * 1024**3)
-    
+
     def test_gb_unit_edge_cases(self):
         with self.assertRaises(ValueError):
             _parse_global_segment_size("GB")
         with self.assertRaises(ValueError):
             _parse_global_segment_size("abcGB")
-    
+
     def test_mb_unit(self):
         self.assertEqual(_parse_global_segment_size("512MB"), 512 * 1024**2)
         self.assertEqual(_parse_global_segment_size("0.5MB"),
                          int(0.5 * 1024**2))
         self.assertEqual(_parse_global_segment_size("1024MB"), 1024 * 1024**2)
-    
+
     def test_kb_unit(self):
         self.assertEqual(_parse_global_segment_size("256KB"), 256 * 1024)
         self.assertEqual(_parse_global_segment_size("1.25KB"),
@@ -36,24 +36,24 @@ def test_kb_unit(self):
     def test_b_unit(self):
         self.assertEqual(_parse_global_segment_size("4096B"), 4096)
         self.assertEqual(_parse_global_segment_size("1024b"), 1024)
-    
+
     def test_no_unit(self):
         self.assertEqual(_parse_global_segment_size("2048"), 2048)
         self.assertEqual(_parse_global_segment_size("0"), 0)
 
     def test_non_string_non_int_input(self):
         self.assertEqual(_parse_global_segment_size(2048.0), 2048)
         self.assertEqual(_parse_global_segment_size(True), 1)
-        
+
         with self.assertRaises(TypeError):
             _parse_global_segment_size(None)
-        
+
         with self.assertRaises(TypeError):
             _parse_global_segment_size({"size": 1024})
 
 
 class TestConvertToBytes(unittest.TestCase):
-    
+
     def test_valid_conversion(self):
         self.assertEqual(_convert_to_bytes("10", 1, "10"), 10)
         self.assertEqual(_convert_to_bytes("1.5", 1024, "1.5KB"),
@@ -63,7 +63,6 @@ def test_valid_conversion(self):
     def test_invalid_numbers(self):
         with self.assertRaises(ValueError):
             _convert_to_bytes("abc", 1, "abc")
-        
+
         with self.assertRaises(ValueError):
             _convert_to_bytes("1.2.3", 1024, "1.2.3KB")
-    

vllm_ascend/distributed/mooncake/config_data.py

@@ -480,7 +480,7 @@ def _parse_global_segment_size(value) -> int:
             raise TypeError(
                 f"Unsupported type for global_segment_size: {type(value)}"
             ) from e
-    
+
     cleaned_input = value.strip().lower()
     if not cleaned_input:
         raise ValueError("global segment size cannot be empty.")
@@ -493,13 +493,13 @@ def _parse_global_segment_size(value) -> int:
     }
     pattern = r'^\s*([\d.]+)\s*(gb|mb|kb|b)?\s*$'
     match = re.match(pattern, cleaned_input)
-    
+
     if not match:
         raise ValueError(f"Invalid format: '{value}'")
-    
+
     number_str = match.group(1)
     unit = match.group(2) or 'b'
-       
+
     multiplier = UNIT_MULTIPLIERS[unit]
     return _convert_to_bytes(number_str, multiplier, value)