From ab0c06dcf3fd1d8ec7e924d7c7672e2a37b15e76 Mon Sep 17 00:00:00 2001
From: Hamza Tahir <hamza@zenml.io>
Date: Fri, 20 Dec 2024 22:21:42 +0100
Subject: [PATCH] Update error handling messages for AWS in Sagemaker
 orchestrator

---
 .../integrations/aws/orchestrators/sagemaker_orchestrator.py  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py b/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py
index 11ee935313..29e0389494 100644
--- a/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py
+++ b/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py
@@ -244,8 +244,8 @@ def prepare_or_run_pipeline(
             TypeError: If the network_config passed is not compatible with the
                 AWS SageMaker NetworkConfig class.
             KeyError: If required fields are missing from schedule_info.
-            ClientError: If there's an AWS API error
-            BotoCoreError: If there's an error in the AWS SDK
+            ClientError: If there's an AWS API error.
+            BotoCoreError: If there's an error in the AWS SDK.
 
         Yields:
             A dictionary of metadata related to the pipeline run.