From ed0ff75c65217be716cefc6b06d5c254efb1dfd8 Mon Sep 17 00:00:00 2001 From: kaiqli <59279714+kaiqli@users.noreply.github.com> Date: Mon, 22 Mar 2021 14:42:19 +0800 Subject: [PATCH] fixed internal logger dumplicated output (#299) * fixed internal logger dumplicated output * delete unused import * fixed isort --- maro/communication/message.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/maro/communication/message.py b/maro/communication/message.py index 59752692..6ae4cdc8 100644 --- a/maro/communication/message.py +++ b/maro/communication/message.py @@ -2,19 +2,12 @@ # Licensed under the MIT license. # native lib -import sys import uuid from enum import Enum from typing import Union -# private lib -from maro.utils import InternalLogger -from maro.utils.exit_code import NON_RESTART_EXIT_CODE - from .utils import session_id_generator -logger = InternalLogger(component_name="message") - class SessionType(Enum): """Communication session categories. @@ -113,5 +106,4 @@ class SessionMessage(Message): elif self.session_type == SessionType.NOTIFICATION: self.session_stage = session_stage if session_stage else NotificationSessionStage.REQUEST else: - logger.error(f"Receive unrecognized session type {self.session_type}, please use the SessionType class.") - sys.exit(NON_RESTART_EXIT_CODE) + raise KeyError(f'Unsupported session type {self.session_type}.')