From 25495723ca9adb8884c2bd171cc4606de57a7e1c Mon Sep 17 00:00:00 2001
From: Gerard Casas Saez <gerardc@squareup.com>
Date: Mon, 1 May 2023 16:06:49 -0600
Subject: [PATCH] Update note on SageMaker MME support for ensemble

---
 src/sagemaker_server.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/sagemaker_server.cc b/src/sagemaker_server.cc
index f18f32c675..7ca77d4960 100644
--- a/src/sagemaker_server.cc
+++ b/src/sagemaker_server.cc
@@ -947,9 +947,9 @@ SagemakerAPIServer::SageMakerMMELoadModel(
   std::string repo_path = parse_map.at("url");
   std::string model_name = parse_map.at("model_name");
 
-  /* Error out if there's more than one subdir/version within
-   * supplied model repo, as ensemble in MME is not (currently)
-   * supported
+  /* Check subdirs for models and find ensemble model within the repo_path
+   * If only 1 model, that will be selected as model_subdir
+   * Else ensemble model directory is set as model_subdir
    */
   DIR* dir;
   struct dirent* ent;