Apply ufmt linter to all py files under tools (#81285)

With ufmt in place https://github.com/pytorch/pytorch/pull/81157, we can now use it to gradually format all files. I'm breaking this down into multiple smaller batches to avoid too many merge conflicts later on.

This batch (as copied from the current BLACK linter config):
* `tools/**/*.py`

Upcoming batchs:
* `torchgen/**/*.py`
* `torch/package/**/*.py`
* `torch/onnx/**/*.py`
* `torch/_refs/**/*.py`
* `torch/_prims/**/*.py`
* `torch/_meta_registrations.py`
* `torch/_decomp/**/*.py`
* `test/onnx/**/*.py`

Once they are all formatted, BLACK linter will be removed.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/81285
Approved by: https://github.com/suo

tools/stats/import_test_stats.py

@@ -5,7 +5,7 @@ import json
 import os
 import pathlib
 import re
-from typing import Any, Callable, Dict, List, Optional, cast
+from typing import Any, Callable, cast, Dict, List, Optional
 from urllib.request import urlopen