Delete the error function usage

Author: Liziqi-77

tests/ut/distributed/mooncake/test_config_data.py

@@ -1,6 +1,7 @@
 import unittest
 
-from vllm_ascend.distributed.mooncake.config_data import _parse_global_segment_size, _convert_to_bytes
+from vllm_ascend.distributed.mooncake.config_data import (
+    _parse_global_segment_size, _convert_to_bytes)
 
 
 class TestParseGlobalSegmentSize(unittest.TestCase):

vllm_ascend/distributed/mooncake/config_data.py

@@ -2,6 +2,7 @@
 import hashlib
 import json
 import os
+from typing import Union
 from dataclasses import dataclass
 from typing import Iterable, List, Optional, Tuple, Union
 
@@ -421,7 +422,7 @@ class LasyerMultiBlockReqMeta:
 class MooncakeStoreConfig:
     local_hostname: str
     metadata_server: str
-    global_segment_size: int | str
+    global_segment_size: Union[int, str]
     local_buffer_size: int
     protocol: str
     device_name: str
@@ -438,9 +439,7 @@ def from_file(file_path: str) -> "MooncakeStoreConfig":
             global_segment_size=_parse_global_segment_size(
                 config.get("global_segment_size", DEFAULT_GLOBAL_SEGMENT_SIZE)
             ),
-            local_buffer_size=_parse_global_segment_size(
-                config.get("local_buffer_size", DEFAULT_LOCAL_BUFFER_SIZE)
-            ),
+            local_buffer_size=config.get("local_buffer_size", DEFAULT_LOCAL_BUFFER_SIZE),
             protocol=config.get("protocol", "tcp"),
             device_name=config.get("device_name", ""),
             master_server_address=config.get("master_server_address"),
@@ -476,13 +475,15 @@ def _parse_global_segment_size(value) -> int:
         try:
             return int(value)
         except (TypeError, ValueError) as e:
-            raise TypeError(f"Unsupported type for global_segment_size: {type(value)}") from e
+            raise TypeError(
+                f"Unsupported type for global_segment_size: {type(value)}"
+            ) from e
     # Clean input string
     cleaned_input = value.strip().lower()
     if not cleaned_input:
         raise ValueError("global segment size cannot be empty.")
 
-    UNIT_MULTIPLIERS  = {
+    UNIT_MULTIPLIERS = {
         'gb': 1024 ** 3,  # 1 GB = 1024^3 bytes
         'mb': 1024 ** 2,  # 1 MB = 1024^2 bytes
         'kb': 1024,       # 1 KB = 1024 bytes
@@ -493,13 +494,16 @@ def _parse_global_segment_size(value) -> int:
         if cleaned_input.endswith(unit):
             number_part = cleaned_input[:-len(unit)].strip()
             if not number_part:
-                raise ValueError(f"Missing numeric value before unit '{unit}' in: '{value}'")
+                raise ValueError(
+                    f"Missing numeric value before unit '{unit}' in: '{value}'"
+                )
             return _convert_to_bytes(number_part, multiplier, value)
     # Handle unit-less input (bytes)
     return _convert_to_bytes(cleaned_input, 1, value)
 
 
-def _convert_to_bytes(number_str: str, multiplier: int, original_input: str) -> int:
+def _convert_to_bytes(number_str: str, multiplier: int, 
+                      original_input: str) -> int:
     """
     Convert numeric string to byte count
     
@@ -517,7 +521,9 @@ def _convert_to_bytes(number_str: str, multiplier: int, original_input: str) ->
     try:
         numeric_value = float(number_str)
     except ValueError:
-        raise ValueError(f"Invalid numeric value '{number_str}' in: '{original_input}'")
+        raise ValueError(
+            f"Invalid numeric value '{number_str}' in: '{original_input}'"
+        )
     # Calculate byte count
     try:
         byte_count = int(numeric_value * multiplier)