From 71d2f4adde9e01b2f45e0b189f5553782cbdfa2b Mon Sep 17 00:00:00 2001 From: you-n-g Date: Thu, 25 Jul 2024 23:12:42 +0800 Subject: [PATCH] loguru mypy error (#118) --- rdagent/log/logger.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/rdagent/log/logger.py b/rdagent/log/logger.py index c35d37fe..fc987fea 100644 --- a/rdagent/log/logger.py +++ b/rdagent/log/logger.py @@ -6,10 +6,12 @@ from functools import partial from multiprocessing import Pipe from multiprocessing.connection import Connection from pathlib import Path -from typing import Union, Generator, Dict, Any +from typing import TYPE_CHECKING, Union, Generator, Dict, Any from logging import LogRecord -from loguru import Record, logger +from loguru import logger +if TYPE_CHECKING: + from loguru import Record from psutil import Process from rdagent.core.conf import RD_AGENT_SETTINGS @@ -97,7 +99,7 @@ class RDAgentLog(SingletonBaseClass): process = parent_process return pid_chain - def file_format(self, record: Record, raw: bool = False) -> str: + def file_format(self, record: "Record", raw: bool = False) -> str: # FIXME: the formmat is tightly coupled with the message reading in storage. record["message"] = LogColors.remove_ansi_codes(record["message"]) if raw: