@@ -49,7 +49,7 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
4949
5050 optimizer_lambda = _lambda .DockerImageFunction (self , "OptimizerFunction" ,
5151 code = _lambda .DockerImageCode .from_ecr (ecr_repository ,
52- tag = image_tag ,
52+ tag_or_digest = image_tag , # Changed from 'tag' to 'tag_or_digest'
5353 cmd = ["lambda_handler.optimizer.handler.lambda_handler" ]
5454 ),
5555 memory_size = 1024 ,
@@ -87,10 +87,7 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
8787 data_source = sfn_tasks .DataSource (
8888 s3_data_source = sfn_tasks .S3DataSource (
8989 s3_data_type = sfn_tasks .S3DataType .S3_PREFIX ,
90- s3_location = sfn_tasks .S3Location (
91- s3_bucket = artifact_bucket ,
92- s3_key = "data/"
93- )
90+ s3_uri = f"s3://{ artifact_bucket .bucket_name } /data/" # Changed from S3Location to s3_uri
9491 )
9592 )
9693 )
@@ -134,4 +131,4 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
134131 events .Rule (self , "WeeklyTriggerRule" ,
135132 schedule = events .Schedule .cron (minute = "0" , hour = "20" , week_day = "WED" ),
136133 targets = [targets .SfnStateMachine (state_machine )]
137- )
134+ )
0 commit comments