From 744205746dcdae6a8efdca231e982e2b5281451e Mon Sep 17 00:00:00 2001 From: papriwal Date: Fri, 5 Dec 2025 00:02:06 -0800 Subject: [PATCH 1/2] Add test dependancy for unit tests in sagemaker-serve --- sagemaker-serve/pyproject.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/sagemaker-serve/pyproject.toml b/sagemaker-serve/pyproject.toml index 43ee46c6b7..f9fc5026ad 100644 --- a/sagemaker-serve/pyproject.toml +++ b/sagemaker-serve/pyproject.toml @@ -43,6 +43,7 @@ test = [ "pytest", "pytest-cov", "mock", + "tensorflow", ] dev = [ "pytest", From 2693e615480cc8e015bbac6125700c12349f4a04 Mon Sep 17 00:00:00 2001 From: papriwal Date: Fri, 5 Dec 2025 00:58:32 -0800 Subject: [PATCH 2/2] Update sagemaker-train test dependancies and fix sagemaker-server unit test --- sagemaker-serve/tests/unit/test_model_builder_servers.py | 2 +- sagemaker-train/pyproject.toml | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/sagemaker-serve/tests/unit/test_model_builder_servers.py b/sagemaker-serve/tests/unit/test_model_builder_servers.py index 9c5c0ef03e..b487f93ba5 100644 --- 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 index 89c7271a2f..1bb0b04a56 100644 --- 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]