minor test fixes

Summary: remove test warnings

Reviewed By: YLGH

Differential Revision: D33835030

fbshipit-source-id: dc8d38972ffbae05bbdf5fcb9a254c955b447f8c

Author: colin2328

torchrec/modules/tests/test_embedding_modules.py

@@ -178,7 +178,7 @@ def test_device(self) -> None:
             name="t1", embedding_dim=3, num_embeddings=10, feature_names=["f1"]
         )
         ebc = EmbeddingBagCollection(tables=[config], device=torch.device("meta"))
-        self.assertEquals(torch.device("meta"), ebc.embedding_bags["t1"].weight.device)
+        self.assertEqual(torch.device("meta"), ebc.embedding_bags["t1"].weight.device)
 
 
 class EmbeddingCollectionTest(unittest.TestCase):
@@ -266,4 +266,4 @@ def test_device(self) -> None:
             name="t1", embedding_dim=3, num_embeddings=10, feature_names=["f1"]
         )
         eb = EmbeddingCollection(tables=[config], device=torch.device("meta"))
-        self.assertEquals(torch.device("meta"), eb.embeddings["t1"].weight.device)
+        self.assertEqual(torch.device("meta"), eb.embeddings["t1"].weight.device)

torchrec/optim/tests/test_warmup.py

@@ -50,7 +50,7 @@ def get_optimizer() -> WarmupOptimizer:
             warmup_optimizer_1.step()
 
         param_state = list(warmup_optimizer_1.state.values())[0]
-        self.assertEquals(
+        self.assertEqual(
             param_state["warmup"].tolist()[0],
             num_iters,
         )