Skip to content

Commit 5e94f09

Browse files
Bob Strahangudivt
authored andcommitted
Simplify UDOP model path construction using f-string
1 parent 1499e04 commit 5e94f09

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)