fbcode//torchx/runner/test

Differential Revision: D67847563
pytorch · Jan 6, 2025 · 04ab5b9 · 04ab5b9
1 parent 5e44f20
commit 04ab5b9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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 = {}