fix missing import

jiapinw · jiapinw · commit bfe2696cfcd4 · 2024-01-18T11:32:08.000-08:00
diff --git a/tests/integ/sagemaker/serve/test_serve_pt_happy.py b/tests/integ/sagemaker/serve/test_serve_pt_happy.py
@@ -28,6 +28,7 @@
 from tests.integ.sagemaker.serve.constants import (
     PYTORCH_SQUEEZENET_RESOURCE_DIR,
     SERVE_SAGEMAKER_ENDPOINT_TIMEOUT,
+    SERVE_LOCAL_CONTAINER_TIMEOUT,
     PYTHON_VERSION_IS_NOT_310,
 )
 from tests.integ.timeout import timeout

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@`
`28`	`28`	`from tests.integ.sagemaker.serve.constants import (`
`29`	`29`	`PYTORCH_SQUEEZENET_RESOURCE_DIR,`
`30`	`30`	`SERVE_SAGEMAKER_ENDPOINT_TIMEOUT,`
	`31`	`+ SERVE_LOCAL_CONTAINER_TIMEOUT,`
`31`	`32`	`PYTHON_VERSION_IS_NOT_310,`
`32`	`33`	`)`
`33`	`34`	`from tests.integ.timeout import timeout`