Remove debugging

octoml · Nov 16, 2023 · cb41ed1 · cb41ed1
1 parent 04f76fd
commit cb41ed1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/serve/mlc_serve/engine/async_connector.py b/serve/mlc_serve/engine/async_connector.py
@@ -17,7 +17,7 @@
 class TextGenerationError(Exception):
     def __init__(self, error: Any) -> None:
         self.error = error
-        super(self).__init__(error)
+        super().__init__(error)
 
 
 class AsyncEngineConnector:
@@ -100,7 +100,6 @@ async def _get_queue_item_until_stopped(self, queue: ResultQueue) -> RequestOutp
 
         if wait_shutdown_task.done():
             if self.engine_loop_exception is not None:
-                import pdb; pdb.set_trace()
                 raise RuntimeError(
                     f"InferenceEngine raised exception: {self.engine_loop_exception}"
                 )

diff --git a/serve/mlc_serve/engine/staging_engine_worker.py b/serve/mlc_serve/engine/staging_engine_worker.py
@@ -15,8 +15,6 @@
 
 logger = logging.getLogger(__name__)
 
-LOG = structlog.stdlib.get_logger(__name__)
-
 @dataclass
 class ShutdownCommand:
     pass
@@ -87,7 +85,6 @@ def add(self, request_states: list[RequestState]):
                     self.cancelled_requests.append(request_state)
                 else:
                     valid_states.append(request_state)
-            LOG.info("valid_states", valid_states=valid_states)
             self.queue.extend(valid_states)
             self.has_new_requests.notify_all()