Skip to content

Commit 41bf134

Browse files
committed
Fix user agent tag issue
1 parent 0e664c8 commit 41bf134

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
@@ -527,6 +527,8 @@ def _user_agent_decorator(self, func):
527527
def wrapper(*args, **kwargs):
528528
# Call the original function
529529
result = func(*args, **kwargs)
530+
if "ModelBuilder" in result:
531+
return result
530532
return result + " ModelBuilder"
531533

532534
return wrapper

0 commit comments

Comments
 (0)