From c82ef4e73366960d0b5782a5f6e54cb7b3cc7a13 Mon Sep 17 00:00:00 2001 From: Dongeon Kim Date: Wed, 13 Dec 2023 18:18:31 +0900 Subject: [PATCH] fix: Fix py38 type union unsupported issue --- alignai/integration/langchain.py | 2 ++ alignai/integration/llama_index.py | 2 ++ alignai/logger.py | 6 +----- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/alignai/integration/langchain.py b/alignai/integration/langchain.py index a01fd76..e0000d0 100644 --- a/alignai/integration/langchain.py +++ b/alignai/integration/langchain.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import importlib from datetime import datetime from importlib import metadata as importlib_metadata diff --git a/alignai/integration/llama_index.py b/alignai/integration/llama_index.py index aaeec41..97de653 100644 --- a/alignai/integration/llama_index.py +++ b/alignai/integration/llama_index.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import importlib from datetime import datetime from importlib import metadata as importlib_metadata diff --git a/alignai/logger.py b/alignai/logger.py index a434c18..6630eec 100644 --- a/alignai/logger.py +++ b/alignai/logger.py @@ -1,7 +1,5 @@ import logging import traceback -import types -from typing import Type class LimitedTracebackFormatter(logging.Formatter): @@ -9,9 +7,7 @@ def __init__(self, *args, stack_trace_limit=5, **kwargs): super().__init__(*args, **kwargs) self._stack_trace_limit = stack_trace_limit - def formatException( - self, exc_info: tuple[Type[BaseException], BaseException, types.TracebackType] | tuple[None, None, None] - ) -> str: + def formatException(self, exc_info) -> str: _, _, tb = exc_info return "".join(traceback.format_tb(tb)[-self._stack_trace_limit :])