Skip to content

Commit ca22f53

Browse files
ruiliann666knikure
authored andcommitted
Fix user agent tag issue
1 parent 3ed0011 commit ca22f53

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
@@ -528,7 +528,10 @@ def _user_agent_decorator(self, func):
528528
def wrapper(*args, **kwargs):
529529
# Call the original function
530530
result = func(*args, **kwargs)
531-
return result + " ModelBuilder"
531+
if "ModelBuilder" in result:
532+
return result
533+
else:
534+
return result + " ModelBuilder"
532535

533536
return wrapper
534537

0 commit comments

Comments
 (0)