Fixed Multiple Args Testing Issue

Differential Revision: D55525717

Pull Request resolved: #868

Author: andywag

torchx/cli/test/cmd_run_test.py

@@ -61,7 +61,7 @@ def test_run_with_multiple_schedule_args(self) -> None:
 
         args = [
             "--scheduler",
-            "local_conda",
+            "local_cwd",
             "--scheduler",
             "local_cwd",
         ]

torchx/cli/test/main_test.py

@@ -13,6 +13,8 @@
 from pathlib import Path
 from unittest.mock import MagicMock, patch
 
+from torchx.cli.argparse_util import ArgOnceAction, torchxconfig
+
 from torchx.cli.cmd_base import SubCommand
 from torchx.cli.main import get_sub_cmds, main
 
@@ -37,6 +39,8 @@ def setUp(self) -> None:
 
     def tearDown(self) -> None:
         os.chdir(self.old_cwd)
+        ArgOnceAction.called_args = set()
+        torchxconfig.called_args = set()
 
     def test_run_abs_config_path(self) -> None:
         main(