Fix Unit tests

dk19y · dk19y · commit bb0acc5cd71a · 2020-09-02T00:29:15.000Z
diff --git a/test/unit/test_model_server.py b/test/unit/test_model_server.py
@@ -150,7 +150,7 @@ def test_adapt_to_ts_format(path_exists, make_dir, subprocess_check_call, set_py
         "--export-path",
         torchserve.DEFAULT_TS_MODEL_DIRECTORY,
         "--extra-files",
-        os.path.join(environment.model_dir, environment.Environment().module_name + ".py"),
+        os.path.join(environment.model_dir, torchserve.DEFAULT_TS_CODE_DIR, environment.Environment().module_name + ".py"),
         "--version",
         "1",
     ]

Original file line number	Diff line number	Diff line change
`@@ -150,7 +150,7 @@ def test_adapt_to_ts_format(path_exists, make_dir, subprocess_check_call, set_py`
`150`	`150`	`"--export-path",`
`151`	`151`	`torchserve.DEFAULT_TS_MODEL_DIRECTORY,`
`152`	`152`	`"--extra-files",`
`153`		`- os.path.join(environment.model_dir, environment.Environment().module_name + ".py"),`
	`153`	`+ os.path.join(environment.model_dir, torchserve.DEFAULT_TS_CODE_DIR, environment.Environment().module_name + ".py"),`
`154`	`154`	`"--version",`
`155`	`155`	`"1",`
`156`	`156`	`]`