fix: add extractedTag for dockerImageCode

aws-solutions-library-samples · Dec 6, 2024 · e1ffe75 · e1ffe75
1 parent 95718e4
commit e1ffe75
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/osml/osml_container.ts b/lib/osml/osml_container.ts
@@ -268,7 +268,9 @@ export class OSMLContainer extends Construct {
 
       if (this.buildDockerImageCode) {
         // Create a DockerImageCode object for Lambda using the imported ECR repository.
-        this.dockerImageCode = DockerImageCode.fromEcr(this.repository);
+        this.dockerImageCode = DockerImageCode.fromEcr(this.repository, {
+          tagOrDigest: extractedTag
+        });
       }
     } else {
       // If the CONTAINER_URI does not indicate an ECR repository, assume it is a public or private Docker registry.