From ccc2be12720eaccccb8616dc477eefda95d02a81 Mon Sep 17 00:00:00 2001
From: CodemodService PythonDeprecatedAssertsBot
<noreply+2443911735787003@fb.com>
Date: Mon, 6 Jan 2025 06:50:52 -0800
Subject: [PATCH] fbcode//torchx/schedulers/test
Differential Revision: D67847525
---
torchx/schedulers/test/aws_batch_scheduler_test.py | 6 +++---
torchx/schedulers/test/kubernetes_scheduler_test.py | 4 ++--
2 files changed, 5 insertions(+), 5 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"},
diff --git a/torchx/schedulers/test/kubernetes_scheduler_test.py b/torchx/schedulers/test/kubernetes_scheduler_test.py
index a03b2834b..6adc5b51b 100644
--- a/torchx/schedulers/test/kubernetes_scheduler_test.py
+++ b/torchx/schedulers/test/kubernetes_scheduler_test.py
@@ -99,8 +99,8 @@ def test_create_scheduler(self) -> None:
docker_client = MagicMock
scheduler = create_scheduler("foo", client=client, docker_client=docker_client)
self.assertIsInstance(scheduler, kubernetes_scheduler.KubernetesScheduler)
- self.assertEquals(scheduler._docker_client, docker_client)
- self.assertEquals(scheduler._client, client)
+ self.assertEqual(scheduler._docker_client, docker_client)
+ self.assertEqual(scheduler._client, client)
def test_app_to_resource_resolved_macros(self) -> None:
app = _test_app()