From 44fa6c5039dc3b5751a08e781e81605c6b05a631 Mon Sep 17 00:00:00 2001 From: Hamza Tahir Date: Mon, 23 Dec 2024 09:16:00 +0100 Subject: [PATCH] Remove redundant code for getting SageMaker session --- .../integrations/aws/orchestrators/sagemaker_orchestrator.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py b/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py index f751db42db..72066fae64 100644 --- a/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py +++ b/src/zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py @@ -248,9 +248,6 @@ def prepare_or_run_pipeline( Yields: A dictionary of metadata related to the pipeline run. """ - # Get the session and client - session = self._get_sagemaker_session() - # sagemaker requires pipelineName to use alphanum and hyphens only unsanitized_orchestrator_run_name = get_orchestrator_run_name( pipeline_name=deployment.pipeline_configuration.name