fix: resolve incorrect variable assignment for SageMaker API call

datahub-project · Jul 22, 2024 · 578a250 · 578a250
1 parent 8b88930
commit 578a250
Showing 1 changed file with 1 addition and 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
@@ -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()
             )