Skip to content

Commit b95330d

Browse files
committed
fix
1 parent 82f3312 commit b95330d

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

tests/ut/distributed/mooncake/test_config_data.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66

77
class TestParseGlobalSegmentSize(unittest.TestCase):
8+
89
def test_int_input(self):
910
self.assertEqual(_parse_global_segment_size(1024), 1024)
1011
self.assertEqual(_parse_global_segment_size(0), 0)
@@ -52,6 +53,7 @@ def test_non_string_non_int_input(self):
5253

5354

5455
class TestConvertToBytes(unittest.TestCase):
56+
5557
def test_valid_conversion(self):
5658
self.assertEqual(_convert_to_bytes("10", 1, "10"), 10)
5759
self.assertEqual(_convert_to_bytes("1.5", 1024, "1.5KB"),

vllm_ascend/distributed/mooncake/config_data.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
DEFAULT_GLOBAL_SEGMENT_SIZE = 3355443200 # 3.125 GiB
1616
DEFAULT_LOCAL_BUFFER_SIZE = 1073741824 # 1.0 GiB
1717

18+
1819
@dataclass
1920
class MooncakeEngineMetadata:
2021
"""name of the LLM model"""

0 commit comments

Comments
 (0)