Update src/sagemaker_pytorch_serving_container/torchserve.py

dk19y · laurenyu · web-flow · commit 292cf5a37b4e · 2020-07-27T12:23:48.000-07:00
Co-authored-by: Lauren Yu &lt;6631887+laurenyu@users.noreply.github.com&gt;
diff --git a/src/sagemaker_pytorch_serving_container/torchserve.py b/src/sagemaker_pytorch_serving_container/torchserve.py
@@ -40,7 +40,7 @@
 DEFAULT_TS_LOG_FILE = pkg_resources.resource_filename(
     sagemaker_pytorch_serving_container.__name__, "/etc/log4j.properties"
 )
-DEFAULT_TS_MODEL_DIRECTORY = os.path.join(os.getcwd(), ".sagemaker/ts/models")
+DEFAULT_TS_MODEL_DIRECTORY = os.path.join(os.getcwd(), ".sagemaker", "ts", "models")
 DEFAULT_TS_MODEL_NAME = "model"
 DEFAULT_TS_MODEL_SERIALIZED_FILE = "model.pth"
 DEFAULT_HANDLER_SERVICE = "sagemaker_pytorch_serving_container.handler_service"

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@`
`40`	`40`	`DEFAULT_TS_LOG_FILE = pkg_resources.resource_filename(`
`41`	`41`	`sagemaker_pytorch_serving_container.__name__, "/etc/log4j.properties"`
`42`	`42`	`)`
`43`		`-DEFAULT_TS_MODEL_DIRECTORY = os.path.join(os.getcwd(), ".sagemaker/ts/models")`
	`43`	`+DEFAULT_TS_MODEL_DIRECTORY = os.path.join(os.getcwd(), ".sagemaker", "ts", "models")`
`44`	`44`	`DEFAULT_TS_MODEL_NAME = "model"`
`45`	`45`	`DEFAULT_TS_MODEL_SERIALIZED_FILE = "model.pth"`
`46`	`46`	`DEFAULT_HANDLER_SERVICE = "sagemaker_pytorch_serving_container.handler_service"`