Fix unit tests

dk19y · dk19y · commit 6489d2f26b83 · 2021-03-11T13:57:13.000Z
diff --git a/test/unit/test_model_server.py b/test/unit/test_model_server.py
@@ -149,6 +149,8 @@ def test_adapt_to_ts_format(path_exists, make_dir, subprocess_check_call, set_py
         torchserve.DEFAULT_TS_MODEL_DIRECTORY,
         "--version",
         "1",
+        "--extra-files",
+        environment.model_dir
     ]
 
     subprocess_check_call.assert_called_once_with(model_archiver_cmd)

Original file line number	Diff line number	Diff line change
`@@ -149,6 +149,8 @@ def test_adapt_to_ts_format(path_exists, make_dir, subprocess_check_call, set_py`
`149`	`149`	`torchserve.DEFAULT_TS_MODEL_DIRECTORY,`
`150`	`150`	`"--version",`
`151`	`151`	`"1",`
	`152`	`+ "--extra-files",`
	`153`	`+ environment.model_dir`
`152`	`154`	`]`
`153`	`155`
`154`	`156`	`subprocess_check_call.assert_called_once_with(model_archiver_cmd)`