From 6532c76e82ba58ebf039ef98108f3cc8a49da4cc Mon Sep 17 00:00:00 2001 From: ZanSara Date: Mon, 29 Jan 2024 14:59:03 +0100 Subject: [PATCH] quotes --- .../amazon_sagemaker/tests/test_sagemaker.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/integrations/amazon_sagemaker/tests/test_sagemaker.py b/integrations/amazon_sagemaker/tests/test_sagemaker.py index cad0624bd..d0f8cf3a7 100644 --- 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)