quotes

deepset-ai · Jan 29, 2024 · 6532c76 · 6532c76
1 parent cc5cdf0
commit 6532c76
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/integrations/amazon_sagemaker/tests/test_sagemaker.py b/integrations/amazon_sagemaker/tests/test_sagemaker.py
@@ -78,16 +78,16 @@ def test_to_from_dict(self, monkeypatch):
         )
         serialized = component.to_dict()
         assert serialized == {
-            'type': 'haystack_integrations.components.generators.amazon_sagemaker.sagemaker.SagemakerGenerator',
-            'init_parameters': {
-                'model': 'test-model',
-                'aws_access_key_id_var': 'MY_ACCESS_KEY_ID',
-                'aws_secret_access_key_var': 'MY_SECRET_ACCESS_KEY',
-                'aws_session_token_var': 'MY_SESSION_TOKEN',
-                'aws_region_name_var': 'MY_REGION',
-                'aws_profile_name_var': 'MY_PROFILE',
-                'aws_custom_attributes': {'custom': 'attr'},
-                'generation_kwargs': {'generation': 'kwargs'},
+            "type": "haystack_integrations.components.generators.amazon_sagemaker.sagemaker.SagemakerGenerator",
+            "init_parameters": {
+                "model": "test-model",
+                "aws_access_key_id_var": "MY_ACCESS_KEY_ID",
+                "aws_secret_access_key_var": "MY_SECRET_ACCESS_KEY",
+                "aws_session_token_var": "MY_SESSION_TOKEN",
+                "aws_region_name_var": "MY_REGION",
+                "aws_profile_name_var": "MY_PROFILE",
+                "aws_custom_attributes": {"custom": "attr"},
+                "generation_kwargs": {"generation": "kwargs"},
             },
         }
         deserialized = SagemakerGenerator.from_dict(serialized)