Update retry.py

langchain-ai · Sep 18, 2024 · e40aabb · e40aabb
1 parent 0bfeb7f
commit e40aabb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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):