frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE][PYFMT] migrate PYFMT for {torch,test}/{nn,optim}/** to ruff format (#144548)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/144548
Approved by: https://github.com/ezyang
This commit is contained in:
Xuehai Pan
2025-06-14 00:48:12 +08:00
committed by PyTorch MergeBot
parent 3e38feb05f
commit 596b418391
65 changed files with 640 additions and 475 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
torch/nn/modules/module.py
View File

@ -412,6 +412,7 @@ class Module:
import torch.nn as nn
import torch.nn.functional as F
class Model(nn.Module):
def __init__(self) -> None:
super().__init__()
@ -1230,16 +1231,13 @@ class Module:
device: Optional[DeviceLikeType] = ...,
dtype: Optional[dtype] = ...,
non_blocking: bool = ...,
) -> Self:
...
) -> Self: ...
@overload
def to(self, dtype: dtype, non_blocking: bool = ...) -> Self:
...
def to(self, dtype: dtype, non_blocking: bool = ...) -> Self: ...
@overload
def to(self, tensor: Tensor, non_blocking: bool = ...) -> Self:
...
def to(self, tensor: Tensor, non_blocking: bool = ...) -> Self: ...
def to(self, *args, **kwargs):
r"""Move and/or cast the parameters and buffers.
@ -1752,7 +1750,11 @@ class Module:
if recording_scopes:
# type ignore was added because at this point one knows that
# torch.jit._trace._trace_module_map is not Optional and has type Dict[Any, Any]
name = torch.jit._trace._trace_module_map[self] if self in torch.jit._trace._trace_module_map else None # type: ignore[index, operator] # noqa: B950
name = (
torch.jit._trace._trace_module_map[self] # type: ignore[index]
if self in torch.jit._trace._trace_module_map # type: ignore[operator]
else None
) # noqa: B950
if name:
tracing_state.push_scope(name)
else:
@ -2161,13 +2163,20 @@ class Module:
@overload
def state_dict(
self, *, destination: T_destination, prefix: str = ..., keep_vars: bool = ...
) -> T_destination:
...
self,
*,
destination: T_destination,
prefix: str = ...,
keep_vars: bool = ...,
) -> T_destination: ...
@overload
def state_dict(self, *, prefix: str = ..., keep_vars: bool = ...) -> dict[str, Any]:
...
def state_dict(
self,
*,
prefix: str = ...,
keep_vars: bool = ...,
) -> dict[str, Any]: ...
# TODO: Change `*args` to `*` and remove the corresponding warning in docs when BC allows.
# Also remove the logic for arg parsing together.