fbcode//torchx/schedulers/test [A]

Differential Revision: D68445042 Pull Request resolved: #1004
pytorch · Jan 31, 2025 · 0467f33 · 0467f33
1 parent ef3b35f
commit 0467f33
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchx/schedulers/test/aws_batch_scheduler_test.py b/torchx/schedulers/test/aws_batch_scheduler_test.py
@@ -756,7 +756,7 @@ def worker() -> None:
 
     def test_resource_requirement_from_resource(self) -> None:
         cpu_resource = Resource(cpu=2, memMB=1024, gpu=0)
-        self.assertEquals(
+        self.assertEqual(
             [
                 {"type": "VCPU", "value": "2"},
                 {"type": "MEMORY", "value": "1024"},
@@ -765,7 +765,7 @@ def test_resource_requirement_from_resource(self) -> None:
         )
 
         zero_cpu_resource = Resource(cpu=0, memMB=1024, gpu=0)
-        self.assertEquals(
+        self.assertEqual(
             [
                 {"type": "VCPU", "value": "1"},
                 {"type": "MEMORY", "value": "1024"},
@@ -774,7 +774,7 @@ def test_resource_requirement_from_resource(self) -> None:
         )
 
         gpu_resource = Resource(cpu=1, memMB=1024, gpu=2)
-        self.assertEquals(
+        self.assertEqual(
             [
                 {"type": "VCPU", "value": "1"},
                 {"type": "MEMORY", "value": "1024"},