Skip to content

Commit b001698

Browse files
authored
fix: Fix user agent tag issue (aws#4289)
Co-authored-by: Ruilian Gao <[email protected]>
1 parent dfbf76e commit b001698

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -528,6 +528,8 @@ def _user_agent_decorator(self, func):
528528
def wrapper(*args, **kwargs):
529529
# Call the original function
530530
result = func(*args, **kwargs)
531+
if "ModelBuilder" in result:
532+
return result
531533
return result + " ModelBuilder"
532534

533535
return wrapper

0 commit comments

Comments
 (0)