apply Black 25.11.0 style in fbcode (78/92)

Summary:
Formats the covered files with pyfmt.

paintitblack

Reviewed By: itamaro

Differential Revision: D90476272

fbshipit-source-id: ff198a4bc3502fa9914722a1c9e2b3d16b8b951c

Author: bowiechen

examples/01_resnet-50/benchmark_ait.py

@@ -17,10 +17,8 @@
 import os
 
 import click
-
 import torch
 from aitemplate.compiler import compile_model, Model
-
 from aitemplate.frontend import Tensor
 from aitemplate.testing import detect_target
 from modeling.resnet import build_resnet_backbone

examples/01_resnet-50/infer_with_torch.py

@@ -17,7 +17,6 @@
 import numpy as np
 import torch
 from aitemplate.compiler import compile_model, Model
-
 from aitemplate.frontend import Tensor
 from aitemplate.testing import detect_target
 from modeling.resnet import build_resnet_backbone

examples/01_resnet-50/modeling/resnet.py

@@ -323,9 +323,9 @@ def make_stage(block_class, num_blocks, *, in_channels, out_channels, **kwargs):
                         f"same length as num_blocks={num_blocks}."
                     )
                     newk = k[: -len("_per_block")]
-                    assert (
-                        newk not in kwargs
-                    ), f"Cannot call make_stage with both {k} and {newk}!"
+                    assert newk not in kwargs, (
+                        f"Cannot call make_stage with both {k} and {newk}!"
+                    )
                     curr_kwargs[newk] = v[i]
                 else:
                     curr_kwargs[k] = v

examples/01_resnet-50/test_correctness.py

@@ -15,9 +15,7 @@
 import unittest
 
 import torch
-
 from aitemplate.compiler import compile_model
-
 from aitemplate.compiler.base import Tensor
 from aitemplate.testing import detect_target
 

examples/02_detectron2/compile_model.py

@@ -16,11 +16,9 @@
 import os
 
 import click
-
 import numpy as np
 import torch
 from aitemplate.compiler import compile_model, Model
-
 from aitemplate.frontend import Tensor
 from aitemplate.testing import detect_target
 from configs import get_cfg_defaults

examples/02_detectron2/demo.py

@@ -70,10 +70,10 @@ def run_model(
     cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = confidence_threshold
     cfg.freeze()
 
-    assert (
-        weight != ""
-    ), "export model first: python convert_pt2ait.py model_d2.pkl params_ait.pkl \
+    assert weight != "", (
+        "export model first: python convert_pt2ait.py model_d2.pkl params_ait.pkl \
     --config configs/faster_rcnn_R_50_DC5.yaml"
+    )
 
     demo = Predictor(cfg)
     print("run {} end2end".format(cfg.MODEL.NAME))

examples/02_detectron2/modeling/backbone/fpn.py

@@ -186,9 +186,9 @@ def _assert_strides_are_log2_contiguous(strides):
     Assert that each stride is 2x times its preceding stride, i.e. "contiguous in log2".
     """
     for i, stride in enumerate(strides[1:], 1):
-        assert (
-            stride == 2 * strides[i - 1]
-        ), "Strides {} {} are not log2 contiguous".format(stride, strides[i - 1])
+        assert stride == 2 * strides[i - 1], (
+            "Strides {} {} are not log2 contiguous".format(stride, strides[i - 1])
+        )
 
 
 class LastLevelMaxPool(nn.Module):

examples/02_detectron2/modeling/backbone/resnet.py

@@ -310,9 +310,9 @@ def make_stage(block_class, num_blocks, *, in_channels, out_channels, **kwargs):
                         f"same length as num_blocks={num_blocks}."
                     )
                     newk = k[: -len("_per_block")]
-                    assert (
-                        newk not in kwargs
-                    ), f"Cannot call make_stage with both {k} and {newk}!"
+                    assert newk not in kwargs, (
+                        f"Cannot call make_stage with both {k} and {newk}!"
+                    )
                     curr_kwargs[newk] = v[i]
                 else:
                     curr_kwargs[k] = v
@@ -421,12 +421,12 @@ def build_resnet_backbone(cfg):
     }[depth]
 
     if depth in [18, 34]:
-        assert (
-            out_channels == 64
-        ), "Must set MODEL.RESNETS.RES2_OUT_CHANNELS = 64 for R18/R34"
-        assert (
-            res5_dilation == 1
-        ), "Must set MODEL.RESNETS.RES5_DILATION = 1 for R18/R34"
+        assert out_channels == 64, (
+            "Must set MODEL.RESNETS.RES2_OUT_CHANNELS = 64 for R18/R34"
+        )
+        assert res5_dilation == 1, (
+            "Must set MODEL.RESNETS.RES5_DILATION = 1 for R18/R34"
+        )
         assert num_groups == 1, "Must set MODEL.RESNETS.NUM_GROUPS = 1 for R18/R34"
 
     stages = []

examples/02_detectron2/modeling/roi_heads/roi_heads.py

@@ -15,7 +15,6 @@
 from typing import Dict
 
 from aitemplate.compiler import ops
-
 from aitemplate.frontend import nn, Tensor
 
 from .box_head import build_box_head, FastRCNNConvFCHead

examples/02_detectron2/predictor/predictor.py

@@ -19,7 +19,6 @@
 import cv2
 import numpy as np
 import torch
-
 from aitemplate.compiler import Model
 from modeling.meta_arch import GeneralizedRCNN
 from PIL import Image