Merge /get_load into /v1/loads (#23010)
This commit is contained in:
@@ -33,7 +33,7 @@ class TestTypeBasedDispatcher(unittest.TestCase):
|
||||
FlushCacheReqInput,
|
||||
FreezeGCReq,
|
||||
GetInternalStateReq,
|
||||
GetLoadReqInput,
|
||||
GetLoadsReqInput,
|
||||
GetWeightsByNameReqInput,
|
||||
InitWeightsSendGroupForRemoteInstanceReqInput,
|
||||
InitWeightsUpdateGroupReqInput,
|
||||
@@ -113,7 +113,7 @@ class TestTypeBasedDispatcher(unittest.TestCase):
|
||||
(ExpertDistributionReq, lambda req: "expert_distribution_handled"),
|
||||
(LoadLoRAAdapterReqInput, lambda req: "load_lora_adapter_handled"),
|
||||
(UnloadLoRAAdapterReqInput, lambda req: "unload_lora_adapter_handled"),
|
||||
(GetLoadReqInput, lambda req: "get_load_handled"),
|
||||
(GetLoadsReqInput, lambda req: "get_loads_handled"),
|
||||
]
|
||||
|
||||
# Create requests that conforms to the real distribution
|
||||
@@ -204,7 +204,7 @@ class TestTypeBasedDispatcher(unittest.TestCase):
|
||||
test_requests.append(GetWeightsByNameReqInput(name=""))
|
||||
test_requests.append(ReleaseMemoryOccupationReqInput())
|
||||
test_requests.append(RpcReqInput(method=""))
|
||||
test_requests.append(GetLoadReqInput())
|
||||
test_requests.append(GetLoadsReqInput())
|
||||
|
||||
dispatcher = TypeBasedDispatcher(mapping)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user