From e40aabb1c1c812b878f16af2e98cefb5e4d29d28 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 18 Sep 2024 10:36:40 -0700 Subject: [PATCH] Update retry.py --- libs/langgraph/langgraph/pregel/retry.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/retry.py b/libs/langgraph/langgraph/pregel/retry.py index 27ff67169..553b25468 100644 --- a/libs/langgraph/langgraph/pregel/retry.py +++ b/libs/langgraph/langgraph/pregel/retry.py @@ -41,7 +41,9 @@ def run_with_retry( if isinstance(retry_policy.retry_on, Sequence): if not isinstance(exc, tuple(retry_policy.retry_on)): raise - elif isinstance(retry_policy.retry_on, type) and issubclass(retry_policy.retry_on, Exception): + elif isinstance(retry_policy.retry_on, type) and issubclass( + retry_policy.retry_on, Exception + ): if not isinstance(exc, retry_policy.retry_on): raise elif callable(retry_policy.retry_on): @@ -105,7 +107,9 @@ async def arun_with_retry( if isinstance(retry_policy.retry_on, Sequence): if not isinstance(exc, tuple(retry_policy.retry_on)): raise - elif isinstance(retry_policy.retry_on, type) and issubclass(retry_policy.retry_on, Exception): + elif isinstance(retry_policy.retry_on, type) and issubclass( + retry_policy.retry_on, Exception + ): if not isinstance(exc, retry_policy.retry_on): raise elif callable(retry_policy.retry_on):