From 53f56e0c7ac50a7cc163cc758cd76c8aabdc1510 Mon Sep 17 00:00:00 2001
From: Yingge He <yinggeh@nvidia.com>
Date: Thu, 11 Dec 2025 08:21:26 -0800
Subject: [PATCH] Add lock_guard

---
 src/sagemaker_server.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sagemaker_server.cc b/src/sagemaker_server.cc
index 49b8a3ba06..0e6dce5ce2 100644
--- a/src/sagemaker_server.cc
+++ b/src/sagemaker_server.cc
@@ -646,6 +646,7 @@ SagemakerAPIServer::SageMakerMMEUnloadModel(
     target_model = model_name_hash;
   }
 
+  std::lock_guard<std::mutex> lock(models_list_mutex_);
   if (sagemaker_models_list_.find(model_name_hash) ==
       sagemaker_models_list_.end()) {
     LOG_VERBOSE(1) << "Model " << target_model << " with model hash "
@@ -740,7 +741,6 @@ SagemakerAPIServer::SageMakerMMEUnloadModel(
 
   TRITONSERVER_ErrorDelete(unregister_err);
 
-  std::lock_guard<std::mutex> lock(models_list_mutex_);
   sagemaker_models_list_.erase(model_name_hash);
 }