Code Quality fix [E721] [1/n] (#3410)

Summary:
Pull Request resolved: #3410

[E721] compare types, for exact checks use `is` / `is not`, for instance checks use `isinstance()`

See https://www.flake8rules.com/rules/E721.htmlArcLint(FLAKE8)

Reviewed By: TroyGarden

Differential Revision: D83733670

fbshipit-source-id: becbcbcb5a39ae88049fcf414c4d7e33c4999577

Author: aliafzal

torchrec/modules/tests/test_lazy_extension.py

@@ -272,7 +272,7 @@ def forward(self, input: torch.Tensor) -> torch.Tensor:
 
         @torch.no_grad()
         def init_weights(m: torch.nn.Module) -> None:
-            if type(m) == TestModule:
+            if isinstance(m, TestModule):
                 m.param.fill_(7.0)
 
         # Case 1: Running `.apply()` without running first forward pass to