Revert "Fix integ tests"

This reverts commit f6513fe.

Author: adishaa

tests/integ/sagemaker/workflow/test_model_create_and_registration.py

@@ -59,7 +59,6 @@
     ConditionGreaterThanOrEqualTo,
     ConditionIn,
 )
-from sagemaker.model_life_cycle import ModelLifeCycle
 from tests.integ.kms_utils import get_or_create_kms_key
 from tests.integ import DATA_DIR
 from sagemaker.model_card import (
@@ -1006,11 +1005,11 @@ def test_model_registration_with_model_life_cycle_object(
         py_version="py3",
         role=role,
     )
-    create_model_life_cycle = ModelLifeCycle(
-        stage="Development",
-        stage_status="In-Progress",
-        stage_description="Development In Progress",
-    )
+    create_model_life_cycle = {
+        "Stage": "Development",
+        "StageStatus": "In-Progress",
+        "StageDescription": "Development In Progress",
+    }
 
     step_register = RegisterModel(
         name="MyRegisterModelStep",

tests/integ/test_model_package.py

@@ -103,14 +103,15 @@ def test_update_model_life_cycle_model_package(sagemaker_session):
         inference_instances=["ml.m5.large"],
         transform_instances=["ml.m5.large"],
         model_package_group_name=model_group_name,
-        model_life_cycle=create_model_life_cycle,
+        model_life_cycle=create_model_life_cycle._to_request_dict(),
     )
 
     desc_model_package = sagemaker_session.sagemaker_client.describe_model_package(
         ModelPackageName=model_package.model_package_arn
     )
+    create_model_life_cycle_req = create_model_life_cycle._to_request_dict()
 
-    assert desc_model_package["ModelLifeCycle"] == create_model_life_cycle
+    assert desc_model_package["ModelLifeCycle"] == create_model_life_cycle_req
 
     update_model_life_cycle = ModelLifeCycle(
         stage="Staging",
@@ -124,7 +125,7 @@ def test_update_model_life_cycle_model_package(sagemaker_session):
     desc_model_package = sagemaker_session.sagemaker_client.describe_model_package(
         ModelPackageName=model_package.model_package_arn
     )
-    assert desc_model_package["ModelLifeCycle"] == update_model_life_cycle
+    assert desc_model_package["ModelLifeCycle"] == update_model_life_cycle_req
 
     sagemaker_session.sagemaker_client.delete_model_package(
         ModelPackageName=model_package.model_package_arn