[4/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort functorch (#127125)

The `usort` config in `pyproject.toml` has no effect due to a typo. Fixing the typo make `usort` do more and generate the changes in the PR. Except `pyproject.toml`, all changes are generated by `lintrunner -a --take UFMT --all-files`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/127125
Approved by: https://github.com/Skylion007
ghstack dependencies: #127122, #127123, #127124
This commit is contained in:
Xuehai Pan
2024-05-25 16:21:09 +00:00
committed by PyTorch MergeBot
parent 35ea5c6b22
commit a28bfb5ed5
47 changed files with 129 additions and 95 deletions

View File

@ -15,8 +15,6 @@ import unittest
import warnings
from functools import partial, wraps
import functorch
# NB: numpy is a testing dependency!
import numpy as np
import torch
@ -24,21 +22,6 @@ import torch.autograd.forward_ad as fwAD
import torch.nn as nn
import torch.nn.functional as F
from common_utils import expectedFailureIf
from functorch import (
combine_state_for_ensemble,
grad,
grad_and_value,
hessian,
jacfwd,
jacrev,
jvp,
make_functional,
make_functional_with_buffers,
make_fx,
vjp,
vmap,
)
from functorch.experimental import functionalize, replace_all_batch_norm_modules_
from torch._C import _ExcludeDispatchKeyGuard, DispatchKey, DispatchKeySet
from torch._dynamo import allow_in_graph
from torch._functorch.eager_transforms import _slice_argnums
@ -81,6 +64,23 @@ from torch.testing._internal.common_utils import (
from torch.utils._pytree import tree_flatten, tree_map, tree_unflatten
import functorch
from functorch import (
combine_state_for_ensemble,
grad,
grad_and_value,
hessian,
jacfwd,
jacrev,
jvp,
make_functional,
make_functional_with_buffers,
make_fx,
vjp,
vmap,
)
from functorch.experimental import functionalize, replace_all_batch_norm_modules_
USE_TORCHVISION = False
try:
import torchvision # noqa: F401