linter

Author: allenwang28

torchft/process_group_test.py

@@ -100,15 +100,12 @@ def run_collective(
     coll = getattr(pg, collective)
     args_list = _build_args(pg=pg, collective=collective, example_tensor=example_tensor)
     works: Dict[str, dist._Work] = {}
+
     def check_tensors(arg: Any) -> None:  # pyre-ignore[2]
         """Recursively check tensors for expected shape and dtype."""
         if isinstance(arg, torch.Tensor):
-            assert (
-                arg.dtype == dtype
-            ), f"Output dtype mismatch: {arg.dtype} != {dtype}"
-            assert (
-                arg.shape == shape
-            ), f"Output shape mismatch: {arg.shape} != {shape}"
+            assert arg.dtype == dtype, f"Output dtype mismatch: {arg.dtype} != {dtype}"
+            assert arg.shape == shape, f"Output shape mismatch: {arg.shape} != {shape}"
         elif isinstance(arg, (list, tuple)):
             for item in arg:
                 check_tensors(item)