add ufmt as code formatter

.circleci/config.yml.in

@@ -190,10 +190,9 @@ jobs:
       - image: circleci/python:3.7
     steps:
       - checkout
-      - run:
-          command: |
-            pip install --user --progress-bar off flake8 typing
-            flake8 --config=setup.cfg .
+      - run: pip install --user --progress-bar=off ufmt>=1.3 usort==0.6.4 black==21.8b0 flake8
+      - run: ufmt check .
+      - run: flake8 --config=setup.cfg .
 
   python_type_check:
     docker:
@@ -207,7 +206,7 @@ jobs:
             pip install --user --progress-bar off mypy
             pip install --user --progress-bar off types-requests
             pip install --user --progress-bar off --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
-            pip install --user --progress-bar off --editable .
+            pip install --user --progress-bar off --no-build-isolation --editable .
             mypy --config-file mypy.ini
 
   docstring_parameters_sync:

.github/workflows/tests-schedule.yml

@@ -30,7 +30,7 @@ jobs:
         run: pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
 
       - name: Install torchvision
-        run: pip install -e .
+        run: pip install --no-build-isolation --editable .
 
       - name: Install all optional dataset requirements
         run: pip install scipy pandas pycocotools lmdb requests

pyproject.toml

@@ -0,0 +1,16 @@
+[tool.usort]
+
+[tool.usort.known]
+
+first_party = ["torch", "torchvision"]
+
+[tool.black]
+
+line-length = 120
+target-version = ["py36"]
+
+[tool.ufmt]
+
+excludes = [
+    "gallery",
+]

setup.cfg

@@ -9,8 +9,14 @@ max-line-length = 120
 
 [flake8]
 max-line-length = 120
-ignore = F401,E402,F403,W503,W504,F821
+ignore = E203, E402, W503, W504, F821
+per-file-ignores =
+    __init__.py: F401, F403, F405
+    ./hubconf.py: F401
+    torchvision/models/mobilenet.py: F401, F403
+    torchvision/models/quantization/mobilenet.py: F401, F403
+    test/smoke_test.py: F401
 exclude = venv
 
 [pydocstyle]
-select = D417 # Missing argument descriptions in the docstring
+select = D417 # Missing argument descriptions in the docstring

torchvision/ops/poolers.py

@@ -2,7 +2,7 @@
 from torch import nn, Tensor
 
 import torchvision
-from torchvision.ops import roi_align
+from .roi_align import roi_align
 from torchvision.ops.boxes import box_area
 
 from typing import Optional, List, Dict, Tuple, Union