Update sagemaker-train test dependancies and fix sagemaker-server unit test

papriwal · papriwal · commit 2693e615480c · 2025-12-05T00:58:32.000-08:00
diff --git a/sagemaker-serve/tests/unit/test_model_builder_servers.py b/sagemaker-serve/tests/unit/test_model_builder_servers.py
@@ -427,7 +427,7 @@ def test_all_supported_model_servers_have_routes(self):
         }
         
         for model_server, method_name in server_method_map.items():
-            with self.subTest(model_server=model_server):
+            with self.subTest():
                 self.mock_builder.model_server = model_server
                 
                 # Mock the specific build method
diff --git a/sagemaker-train/pyproject.toml b/sagemaker-train/pyproject.toml
@@ -57,7 +57,8 @@ test = [
     "pandas",
     "scipy",
     "omegaconf",
-    "graphene"
+    "graphene",
+    "IPython",
 ]
 
 [tool.setuptools.packages.find]

Original file line number	Diff line number	Diff line change
`@@ -427,7 +427,7 @@ def test_all_supported_model_servers_have_routes(self):`
`427`	`427`	`}`
`428`	`428`
`429`	`429`	`for model_server, method_name in server_method_map.items():`
`430`		`- with self.subTest(model_server=model_server):`
	`430`	`+ with self.subTest():`
`431`	`431`	`self.mock_builder.model_server = model_server`
`432`	`432`
`433`	`433`	`# Mock the specific build method`
Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,8 @@ test = [`
`57`	`57`	`"pandas",`
`58`	`58`	`"scipy",`
`59`	`59`	`"omegaconf",`
`60`		`- "graphene"`
	`60`	`+ "graphene",`
	`61`	`+ "IPython",`
`61`	`62`	`]`
`62`	`63`
`63`	`64`	`[tool.setuptools.packages.find]`