fix(ingest/sagemaker): resolve incorrect variable assignment for Sage…

…Maker API call (datahub-project#10965)
aviv-julienjehannet · Jul 25, 2024 · bff99b1 · bff99b1
1 parent 88ef1c3
commit bff99b1
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()
             )