fix for integ

selvask-aws · selvask-aws · commit 1b6f33e5bd33 · 2024-07-19T00:57:23.000+05:30
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -1807,6 +1807,7 @@ def register(
             model_card = ModelCard(
                 name="estimator_card",
                 training_details=training_details,
+                sagemaker_session=self.sagemaker_session,
             )
         return model.register(
             content_types,

Original file line number	Diff line number	Diff line change
`@@ -1807,6 +1807,7 @@ def register(`
`1807`	`1807`	`model_card = ModelCard(`
`1808`	`1808`	`name="estimator_card",`
`1809`	`1809`	`training_details=training_details,`
	`1810`	`+ sagemaker_session=self.sagemaker_session,`
`1810`	`1811`	`)`
`1811`	`1812`	`return model.register(`
`1812`	`1813`	`content_types,`