Fix unit and integ tests

Author: aviruthen

sagemaker-core/pyproject.toml

@@ -9,7 +9,7 @@ description = "An python package for sagemaker core functionalities"
 authors = [
   {name = "AWS", email = "[email protected]"}
 ]
-readme = "README.rst"
+readme = "README.rst" 
 dependencies = [
   # Add your dependencies here (Include lower and upper bounds as applicable)
     "boto3>=1.42.2,<2.0.0",

sagemaker-mlops/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 name = "sagemaker-mlops"
 dynamic = ["version"]
 description = "SageMaker MLOps package for workflow orchestration and model building"
-readme = "README.md"
+readme = "README.md" 
 requires-python = ">=3.9"
 authors = [
     {name = "Amazon Web Services"},

sagemaker-serve/tests/integ/test_tei_integration.py

@@ -99,12 +99,13 @@ def build_and_deploy():
     )
 
     # Build and deploy your model. Returns SageMaker Core Model and Endpoint objects
-    core_model = model_builder.build(model_name=f"{MODEL_NAME_PREFIX}-{unique_id}")
+    core_model = model_builder.build(model_name=f"{MODEL_NAME_PREFIX}-{unique_id}", region=AWS_REGION)
     logger.info(f"Model Successfully Created: {core_model.model_name}")
 
     core_endpoint = model_builder.deploy(
         endpoint_name=f"{ENDPOINT_NAME_PREFIX}-{unique_id}",
-        initial_instance_count=1
+        initial_instance_count=1,
+        region=AWS_REGION
     )
     logger.info(f"Endpoint Successfully Created: {core_endpoint.endpoint_name}")
 

sagemaker-serve/tests/integ/test_tgi_integration.py

@@ -107,12 +107,13 @@ def build_and_deploy():
     )
 
     # Build and deploy your model. Returns SageMaker Core Model and Endpoint objects
-    core_model = model_builder.build(model_name=f"{MODEL_NAME_PREFIX}-{unique_id}")
+    core_model = model_builder.build(model_name=f"{MODEL_NAME_PREFIX}-{unique_id}", region=AWS_REGION)
     logger.info(f"Model Successfully Created: {core_model.model_name}")
 
     core_endpoint = model_builder.deploy(
         endpoint_name=f"{ENDPOINT_NAME_PREFIX}-{unique_id}",
-        initial_instance_count=1
+        initial_instance_count=1,
+        region=AWS_REGION
     )
     logger.info(f"Endpoint Successfully Created: {core_endpoint.endpoint_name}")
 

sagemaker-serve/tox.ini

@@ -90,6 +90,7 @@ commands =
     pip install 'torch==2.3.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'torchvision==0.18.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'dill>=0.3.9'
+    pip install 'tensorflow==2.16.2'
 
     pytest {posargs}
 deps =

sagemaker-train/pyproject.toml

@@ -57,7 +57,8 @@ test = [
     "pandas",
     "scipy",
     "omegaconf",
-    "graphene"
+    "graphene",
+    "IPython"
 ]
 
 [tool.setuptools.packages.find]