Fix path

dk19y · dk19y · commit 395eba3ce2c0 · 2021-03-11T13:44:41.000Z
diff --git a/src/sagemaker_pytorch_serving_container/torchserve.py b/src/sagemaker_pytorch_serving_container/torchserve.py
@@ -120,6 +120,8 @@ def _adapt_to_ts_format(handler_service):
         DEFAULT_TS_MODEL_DIRECTORY,
         "--version",
         "1",
+        "--extra-files",
+        os.path.join(environment.model_dir)
     ]
 
     logger.info(model_archiver_cmd)

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,8 @@ def _adapt_to_ts_format(handler_service):`
`120`	`120`	`DEFAULT_TS_MODEL_DIRECTORY,`
`121`	`121`	`"--version",`
`122`	`122`	`"1",`
	`123`	`+ "--extra-files",`
	`124`	`+ os.path.join(environment.model_dir)`
`123`	`125`	`]`
`124`	`126`
`125`	`127`	`logger.info(model_archiver_cmd)`