From bff99b17f496b263bec2cb918a4ca5fcd7787ced Mon Sep 17 00:00:00 2001 From: Tristan Heisler Date: Tue, 23 Jul 2024 06:27:52 -0600 Subject: [PATCH] fix(ingest/sagemaker): resolve incorrect variable assignment for SageMaker API call (#10965) --- .../datahub/ingestion/source/aws/sagemaker_processors/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py b/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py index e82cfc58f75a74..eef2b26ee08f2e 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py +++ b/metadata-ingestion/src/datahub/ingestion/source/aws/sagemaker_processors/models.py @@ -418,7 +418,7 @@ def get_model_wu( model_image_groups: Set[str] = set() if model_image is not None: - model_uri_groups = self.lineage.model_image_to_groups.get( + model_image_groups = self.lineage.model_image_to_groups.get( model_image, set() )