Skip to content

Commit 9682dcf

Browse files
committed
Fix user agent tag issue
1 parent 0e664c8 commit 9682dcf

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -527,7 +527,10 @@ def _user_agent_decorator(self, func):
527527
def wrapper(*args, **kwargs):
528528
# Call the original function
529529
result = func(*args, **kwargs)
530-
return result + " ModelBuilder"
530+
if "ModelBuilder" in result:
531+
return result
532+
else:
533+
return result + " ModelBuilder"
531534

532535
return wrapper
533536

0 commit comments

Comments
 (0)