mock aws

wandb · Dec 20, 2024 · 0c7b415 · 0c7b415
1 parent 6f205bc
commit 0c7b415
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/integrations/bedrock/bedrock_test.py b/tests/integrations/bedrock/bedrock_test.py
@@ -2,6 +2,7 @@
 import boto3
 import botocore
 import pytest
+from moto import mock_aws
 from unittest.mock import patch
 
 import weave
@@ -99,6 +100,7 @@ def __iter__(self):
 
 
 @pytest.mark.skip_clickhouse_client
+@mock_aws
 def test_bedrock_converse(client: weave.trace.weave_client.WeaveClient) -> None:
     bedrock_client = boto3.client("bedrock-runtime")
     patch_client(bedrock_client)
@@ -148,6 +150,7 @@ def test_bedrock_converse(client: weave.trace.weave_client.WeaveClient) -> None:
 
 
 @pytest.mark.skip_clickhouse_client
+@mock_aws
 def test_bedrock_converse_stream(client: weave.trace.weave_client.WeaveClient) -> None:
     bedrock_client = boto3.client("bedrock-runtime")
     patch_client(bedrock_client)