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

This commit is contained in:
Yuhao Yang
2026-01-04 22:05:39 -08:00
committed by GitHub
parent 2138ff48c6
commit 10174e1114
@@ -8,7 +8,6 @@ import copy
import dataclasses import dataclasses
import logging import logging
import os import os
import pickle
import signal import signal
import sys import sys
import threading import threading
@@ -801,9 +800,7 @@ class GrpcRequestManager:
async def _send_to_scheduler(self, obj): async def _send_to_scheduler(self, obj):
"""Send an object to the scheduler via ZMQ.""" """Send an object to the scheduler via ZMQ."""
try: try:
self.send_to_scheduler.send_multipart( self.send_to_scheduler.send_pyobj(obj)
[b"NORM", pickle.dumps(obj)], copy=False
)
except Exception as e: except Exception as e:
logger.error(f"Failed to send to scheduler: {e}") logger.error(f"Failed to send to scheduler: {e}")
raise raise