From 3879ed61300c051b3bf294177422ade776bf0b03 Mon Sep 17 00:00:00 2001 From: Itamar Oren Date: Wed, 29 Jan 2025 09:29:04 -0800 Subject: [PATCH] fbcode//torchx/schedulers/test [A] Differential Revision: D68445042 --- torchx/schedulers/test/aws_batch_scheduler_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchx/schedulers/test/aws_batch_scheduler_test.py b/torchx/schedulers/test/aws_batch_scheduler_test.py index 482116eb8..13bc2f126 100644 --- 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"},