Skip to content

Commit cc125e1

Browse files
author
tanqingshan
committed
change for ci
Signed-off-by: <> Signed-off-by: tanqingshan <50050625@china.huawei.com>
1 parent 1be62eb commit cc125e1

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tests/ut/eplb/core/policy/test_policy_dynamic_ep_v2.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from typing import Any
2+
13
import numpy as np
24
import pytest
35

@@ -79,7 +81,7 @@ def test_prepare_expert_list():
7981
def test_non_redundant_expert_information():
8082
origin_deployment = np.array([[0, 1]])
8183
updated_weights = [(0, 10), (1, 5)]
82-
rendun_pos = [[]]
84+
rendun_pos: dict[Any, Any] = {}
8385
assignments, weights, loads, counts = DynamicEplbV2.non_redundant_expert_information(
8486
origin_deployment, updated_weights, rendun_pos)
8587
assert assignments[0] == [0, 1]

tests/ut/eplb/core/test_eplb_device_transfer_loader.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from typing import Any
12
from unittest.mock import MagicMock, patch
23

34
import pytest
@@ -103,7 +104,7 @@ def test_invalid_state_asyn_update(mock_adaptor):
103104
loader_obj = loader.D2DExpertWeightLoader(mock_adaptor)
104105

105106
loader_obj.state = loader.ExpertWeightUpdateState.WAITING
106-
reqs = []
107+
reqs: list[Any] = []
107108
loader_obj.asyn_expert_weight_transfer(reqs)
108109
assert reqs == []
109110

0 commit comments

Comments
 (0)