Skip to content

Commit b4a7f5f

Browse files
committed
Merge branch 'fix/udop-param-default' into 'develop'
Simplify UDOP model path construction and avoid invalid default for regions other than us-east-1 and us-west-2 See merge request genaiic-reusable-assets/engagement-artifacts/genaiic-idp-accelerator!284
2 parents 63f08a9 + b94081b commit b4a7f5f

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

publish.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -282,12 +282,7 @@ def setup_environment(self):
282282
self.bucket = f"{self.bucket_basename}-{self.region}"
283283

284284
# Set UDOP model path based on region
285-
if self.region == "us-east-1":
286-
self.public_sample_udop_model = "s3://aws-ml-blog-us-east-1/artifacts/genai-idp/udop-finetuning/rvl-cdip/model.tar.gz"
287-
elif self.region == "us-west-2":
288-
self.public_sample_udop_model = "s3://aws-ml-blog-us-west-2/artifacts/genai-idp/udop-finetuning/rvl-cdip/model.tar.gz"
289-
else:
290-
self.public_sample_udop_model = ""
285+
self.public_sample_udop_model = f"s3://aws-ml-blog-{self.region}/artifacts/genai-idp/udop-finetuning/rvl-cdip/model.tar.gz"
291286

292287
def check_prerequisites(self):
293288
"""Check for required commands and versions"""

0 commit comments

Comments
 (0)