diff --git a/torchx/runner/test/api_test.py b/torchx/runner/test/api_test.py index 67182f38..d488eb3f 100644 --- a/torchx/runner/test/api_test.py +++ b/torchx/runner/test/api_test.py @@ -469,7 +469,7 @@ def test_status_ui_url(self, json_dumps_mock: MagicMock, _) -> None: ) app_handle = runner.run(AppDef(app_id, roles=[role]), scheduler="local_dir") status = none_throws(runner.status(app_handle)) - self.assertEquals(resp.ui_url, status.ui_url) + self.assertEqual(resp.ui_url, status.ui_url) @patch("json.dumps") def test_status_structured_msg(self, json_dumps_mock: MagicMock, _) -> None: @@ -493,7 +493,7 @@ def test_status_structured_msg(self, json_dumps_mock: MagicMock, _) -> None: ) app_handle = runner.run(AppDef(app_id, roles=[role]), scheduler="local_dir") status = none_throws(runner.status(app_handle)) - self.assertEquals(resp.structured_error_msg, status.structured_error_msg) + self.assertEqual(resp.structured_error_msg, status.structured_error_msg) def test_wait_unknown_app(self, _) -> None: with self.get_runner() as runner: diff --git a/torchx/runner/test/config_test.py b/torchx/runner/test/config_test.py index 959c3cf7..cf3b216c 100644 --- a/torchx/runner/test/config_test.py +++ b/torchx/runner/test/config_test.py @@ -425,7 +425,7 @@ def test_load_invalid_runopt(self, _) -> None: # this makes things super hard to guarantee BC - stale config file will fail # to run, we don't want that) - self.assertEquals("option_that_exists", cfg.get("s")) + self.assertEqual("option_that_exists", cfg.get("s")) def test_load_no_section(self) -> None: cfg = {}