Consolidated mock exception

run-llama · Oct 6, 2024 · 0ae859b · 0ae859b
1 parent 9b803c6
commit 0ae859b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/...x-integrations/llms/llama-index-llms-bedrock-converse/tests/test_llms_bedrock_converse.py b/...x-integrations/llms/llama-index-llms-bedrock-converse/tests/test_llms_bedrock_converse.py
@@ -20,12 +20,9 @@
 prompt = "Test"
 
 
-class MockThrottlingException(Exception):
-    pass
-
-
 class MockExceptions:
-    ThrottlingException = MockThrottlingException
+    class ThrottlingException(Exception):
+        pass
 
 
 class AsyncMockClient: