Update tests/integration/cli/test_cli.py

Co-authored-by: Eunwoo Shin <eunwoo.shin@intel.com>
openvinotoolkit · Mar 30, 2023 · bb6f355 · bb6f355
1 parent 16e33f1
commit bb6f355
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/cli/test_cli.py b/tests/integration/cli/test_cli.py
@@ -150,7 +150,7 @@ def test_otx_train_wo_output_args(self, tmp_dir_path):
         assert os.path.exists(os.path.join(output_path, "latest_trained_model", "models"))
         assert os.path.exists(os.path.join(output_path, "latest_trained_model", "models", "weights.pth"))
         assert os.path.exists(os.path.join(output_path, "latest_trained_model", "models", "label_schema.json"))
-        file_list = file_list[:-1]  # Remove latest from list, then file_list[-1] is latest
+        file_list.remove("latest_trained_model")
         assert os.path.exists(os.path.join(output_path, file_list[-1]))
         assert os.path.exists(os.path.join(output_path, file_list[-1], "models"))
         assert os.path.exists(os.path.join(output_path, file_list[-1], "models", "weights.pth"))