fix: ModelCard Object integ Test fix

selvask-aws · selvask-aws · commit c6981e238736 · 2024-06-13T21:54:11.000+05:30
diff --git a/tests/integ/sagemaker/workflow/test_model_create_and_registration.py b/tests/integ/sagemaker/workflow/test_model_create_and_registration.py
@@ -825,6 +825,7 @@ def test_model_registration_with_model_card_object(
 
     my_card = ModelCard(
         name="TestName",
+        sagemaker_session=sagemaker_session_for_pipeline,
         status=ModelCardStatusEnum.DRAFT,
         model_overview=model_overview,
         intended_uses=intended_uses,
