Add test dependancy for unit tests in sagemaker-serve

papriwal · papriwal · commit 744205746dcd · 2025-12-05T00:02:06.000-08:00
diff --git a/sagemaker-serve/pyproject.toml b/sagemaker-serve/pyproject.toml
@@ -43,6 +43,7 @@ test = [
     "pytest",
     "pytest-cov",
     "mock",
+    "tensorflow",
 ]
 dev = [
     "pytest",
