Revert "[grpc] update api to scheduler in grpc request manager" (#16387)

This commit is contained in:
Yuhao Yang
2026-01-05 14:05:39 +08:00
committed by GitHub
parent 2138ff48c6
commit 10174e1114

View File

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