Fixed test cases

awslabs · Aug 29, 2024 · 0245f95 · 0245f95
1 parent f89146e
commit 0245f95
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/modules/examples/mlops-stepfunctions/tests/test_stack.py b/modules/examples/mlops-stepfunctions/tests/test_stack.py
@@ -28,17 +28,15 @@ def stack_model_package_input() -> cdk.Stack:
     module_name = "test-module"
 
     app_prefix = f"{project_name}-{deployment_name}-{module_name}"
-    bucket_policy_arn = "arn:aws:iam::123456789012:policy/bucketPolicy"
-    permission_boundary_arn = "arn:aws:iam::123456789012:policy/boundary"
 
     return stack.MLOPSSFNResources(
         scope=app,
         id=app_prefix,
         project_name=project_name,
         deployment_name=deployment_name,
         module_name=module_name,
-        bucket_policy_arn=bucket_policy_arn,
-        permission_boundary_arn=permission_boundary_arn,
+        model_name="demo",
+        hours="18",
         env=cdk.Environment(
             account=os.environ["CDK_DEFAULT_ACCOUNT"],
             region=os.environ["CDK_DEFAULT_REGION"],