[grpc] update api to scheduler in grpc request manager (#16350)

This commit is contained in:
Chang Su
2026-01-03 10:54:31 -08:00
committed by GitHub
parent 877c8e3a96
commit 7bc13c906c

View File

@@ -8,6 +8,7 @@ import copy
import dataclasses
import logging
import os
import pickle
import signal
import sys
import threading
@@ -800,7 +801,9 @@ class GrpcRequestManager:
async def _send_to_scheduler(self, obj):
"""Send an object to the scheduler via ZMQ."""
try:
self.send_to_scheduler.send_pyobj(obj)
self.send_to_scheduler.send_multipart(
[b"NORM", pickle.dumps(obj)], copy=False
)
except Exception as e:
logger.error(f"Failed to send to scheduler: {e}")
raise