Swap second TimeoutError to TimeoutException as well

leondz · Sep 30, 2024 · c090a4d · c090a4d
1 parent 7935a22
commit c090a4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/garak/generators/ollama.py b/garak/generators/ollama.py
@@ -60,7 +60,7 @@ class OllamaGeneratorChat(OllamaGenerator):
 
     @backoff.on_exception(
         backoff.fibo,
-        (TimeoutError, ollama.ResponseError),
+        (TimeoutException, ollama.ResponseError),
         max_value=70,
         giveup=_give_up,
     )