Flip default value for mypy disallow_untyped_defs [8/11] (#127845)

See #127836 for details.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/127845
Approved by: https://github.com/oulgen
ghstack dependencies: #127842, #127843, #127844

torch/nn/backends/thnn.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # this is for historical pickle deserialization, it is not used otherwise
 
 def _get_thnn_function_backend():

torch/nn/cpp.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Functionality for Python <-> C++ frontend inter-op."""
 
 from torch import nn

torch/nn/grad.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Gradient interface."""
 
 import torch

torch/nn/init.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file contains utilities for initializing neural network parameters."""
 import math
 import warnings

torch/nn/modules/_functions.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.distributed as dist
 

torch/nn/modules/activation.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import warnings
 from typing import Optional, Tuple
 

torch/nn/modules/adaptive.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 from collections import namedtuple
 

torch/nn/modules/batchnorm.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Optional, Any
 
 import torch

torch/nn/modules/container.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import OrderedDict, abc as container_abcs
 from itertools import chain, islice
 import operator

torch/nn/modules/conv.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import math
 
 import torch

torch/nn/modules/flatten.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .module import Module
 
 from typing import Tuple, Union

torch/nn/modules/instancenorm.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 import warnings
 from torch import Tensor

torch/nn/modules/lazy.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import itertools
 from typing import Protocol, Optional, Type, Any
 

torch/nn/modules/linear.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import math
 from typing import Any
 

torch/nn/modules/loss.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .distance import PairwiseDistance
 from .module import Module
 from .. import functional as F

torch/nn/modules/module.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import OrderedDict, namedtuple
 import itertools
 import warnings

torch/nn/modules/normalization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import numbers
 from torch.nn.parameter import Parameter

torch/nn/modules/padding.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .module import Module
 from .utils import _pair, _quadruple, _ntuple
 from .. import functional as F

torch/nn/modules/rnn.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import math
 import warnings
 import numbers

torch/nn/modules/sparse.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Optional
 
 import torch

torch/nn/modules/transformer.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import copy
 from typing import Optional, Any, Union, Callable
 

torch/nn/modules/upsampling.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .module import Module
 from .. import functional as F
 

torch/nn/modules/utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import collections
 from itertools import repeat
 from typing import List, Dict, Any

torch/nn/parallel/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing_extensions import deprecated
 
 from .parallel_apply import parallel_apply

torch/nn/parallel/comm.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import warnings
 import torch
 from torch.cuda import nccl

torch/nn/parallel/data_parallel.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import operator
 import torch
 import warnings

torch/nn/parallel/distributed.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import copy
 import functools
 import inspect

torch/nn/parallel/scatter_gather.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from typing import Any, Dict, List, Optional, Sequence, Tuple, TypeVar, Union, overload
 from typing_extensions import deprecated

torch/nn/parameter.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import builtins
 from typing import Optional, Tuple
 

torch/nn/utils/_deprecation_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import List, Callable
 import importlib
 import warnings

torch/nn/utils/_expanded_weights/conv_expanded_weights.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn.functional as F
 

torch/nn/utils/_expanded_weights/conv_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn.functional as F
 

torch/nn/utils/_expanded_weights/embedding_expanded_weights.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn.functional as F
 from .expanded_weights_impl import implements_per_sample_grads

torch/nn/utils/_expanded_weights/expanded_weights_impl.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from contextlib import contextmanager
 
 import torch

torch/nn/utils/_expanded_weights/expanded_weights_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Optional
 
 import torch

torch/nn/utils/_expanded_weights/group_norm_expanded_weights.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from functools import reduce
 import operator
 import torch

torch/nn/utils/_expanded_weights/instance_norm_expanded_weights.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from functools import partial
 import torch
 import torch.nn.functional as F

torch/nn/utils/_expanded_weights/layer_norm_expanded_weights.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 import torch
 import torch.nn.functional as F

torch/nn/utils/_expanded_weights/linear_expanded_weights.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn.functional as F
 from .expanded_weights_impl import implements_per_sample_grads

torch/nn/utils/_per_sample_grad.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import functools
 
 import torch

torch/nn/utils/clip_grad.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import functools
 from typing import Union, Iterable, List, Dict, Tuple, Optional, cast
 from typing_extensions import deprecated

torch/nn/utils/init.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import inspect
 import torch
 

torch/nn/utils/memory_format.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 

torch/nn/utils/parametrizations.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from enum import Enum, auto
 
 import torch

torch/nn/utils/parametrize.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.__future__ import get_swap_module_params_on_conversion
 from torch.nn.modules.container import ModuleList, ModuleDict, Module

torch/nn/utils/prune.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""Pruning methods."""
 import numbers
 from abc import ABC, abstractmethod

torch/nn/utils/rnn.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Any, Iterable, NamedTuple, Optional, overload, Sequence, Tuple, Union
 
 from typing_extensions import Self

torch/nn/utils/spectral_norm.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Spectral Normalization from https://arxiv.org/abs/1802.05957."""
 import torch
 from torch.nn.functional import normalize

torch/nn/utils/stateless.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import contextlib
 from collections import defaultdict
 from typing import Any, Dict, Iterator, Optional, Set, Tuple, Union

torch/nn/utils/weight_norm.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""Weight Normalization from https://arxiv.org/abs/1602.07868."""
 from torch.nn.parameter import Parameter, UninitializedParameter
 from torch import _weight_norm, norm_except_dim

torch/onnx/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch import _C
 from torch._C import _onnx as _C_onnx
 from torch._C._onnx import (

torch/onnx/_deprecation.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Utility for deprecating functions."""
 
 import functools

torch/onnx/_globals.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Globals used internally by the ONNX exporter.
 
 Do not use this module outside of `torch.onnx` and its tests.

torch/onnx/_internal/_beartype.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """An internal wrapper for the beartype library.
 
 The module returns a no-op decorator when the beartype library is not installed.

torch/onnx/_internal/diagnostics/_diagnostic.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Diagnostic components for TorchScript based ONNX export, i.e. `torch.onnx.export`."""
 from __future__ import annotations
 

torch/onnx/_internal/diagnostics/_rules.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 GENERATED CODE - DO NOT EDIT DIRECTLY
 This file is generated by gen_diagnostics.py.

torch/onnx/_internal/diagnostics/infra/_infra.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file defines an additional layer of abstraction on top of the SARIF OM."""
 
 from __future__ import annotations

torch/onnx/_internal/diagnostics/infra/context.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """A diagnostic context based on SARIF."""
 
 from __future__ import annotations

torch/onnx/_internal/diagnostics/infra/decorator.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import functools

torch/onnx/_internal/exporter.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import (  # for onnx.ModelProto (ONNXProgram) and onnxruntime (ONNXRuntimeOptions)
     annotations,
 )

torch/onnx/_internal/fx/_pass.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import abc

torch/onnx/_internal/fx/analysis/unsupported_nodes.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import dataclasses

torch/onnx/_internal/fx/decomposition_skip.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """A context manager that disables the decomposition of certain ops during dynamo tracing.
 
 The approach is to temporarily hijack the operator callable with PT2 custom operator.

torch/onnx/_internal/fx/decomposition_table.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Dispatcher for AtenLib functions from onnx-script."""
 
 from __future__ import annotations

torch/onnx/_internal/fx/diagnostics.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import dataclasses

torch/onnx/_internal/fx/dynamo_graph_extractor.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # NOTE: This file is referenced by name at
 #       /opt/pytorch/torch/_dynamo/eval_frame.py::DONT_WRAP_FILES.
 #       introduced by https://github.com/pytorch/pytorch/pull/98894.

torch/onnx/_internal/fx/fx_onnx_interpreter.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import inspect

torch/onnx/_internal/fx/fx_symbolic_graph_extractor.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import functools

torch/onnx/_internal/fx/onnxfunction_dispatcher.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Dispatcher for AtenLib functions from onnx-script."""
 
 from __future__ import annotations

torch/onnx/_internal/fx/op_validation.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Module for handling op-level validation during exporting."""
 
 from __future__ import annotations

torch/onnx/_internal/fx/passes/_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Common utility functions for FX passes.
 
 These functions should NOT be directly invoked outside of `passes` package.

torch/onnx/_internal/fx/passes/decomp.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import contextlib

torch/onnx/_internal/fx/passes/functionalization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import contextlib

torch/onnx/_internal/fx/passes/modularization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import abc

torch/onnx/_internal/fx/passes/readability.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 from typing import Dict, List, Sequence, Tuple, Union

torch/onnx/_internal/fx/passes/type_promotion.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Owner(s): ["module: onnx"]
 from __future__ import annotations
 

torch/onnx/_internal/fx/passes/virtualization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 from typing import List, Optional, Tuple

torch/onnx/_internal/fx/patcher.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import copy
 import functools
 import io

torch/onnx/_internal/fx/serialization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import io

torch/onnx/_internal/fx/torch_export_graph_extractor.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # NOTE: This file is referenced by name at
 #       /opt/pytorch/torch/_dynamo/eval_frame.py::DONT_WRAP_FILES.
 #       introduced by https://github.com/pytorch/pytorch/pull/98894.

torch/onnx/_internal/fx/type_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Utilities for converting and operating on ONNX, JIT and torch types."""
 from __future__ import annotations
 

torch/onnx/_internal/io_adapter.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import inspect

torch/onnx/_internal/jit_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Utilities for manipulating the torch.Graph object and the torchscript."""
 from __future__ import annotations
 

torch/onnx/_internal/onnx_proto_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Utilities for manipulating the onnx and onnx-script dependencies and ONNX proto."""
 
 from __future__ import annotations

torch/onnx/_internal/onnxruntime.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import dataclasses
 import importlib
 import logging

torch/onnx/_internal/registration.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Module for handling symbolic function registration."""
 
 import warnings

torch/onnx/_onnx_supported_ops.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import inspect
 from typing import Dict, List, Union
 

torch/onnx/_type_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Utilities for converting and operating on ONNX, JIT and torch types."""
 from __future__ import annotations
 

torch/onnx/operators.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""This file provides a location for operators that help exporting models via onnx.
 
 E.g. `shape_as_tensor` and `reshape_from_tensor_shape`

torch/onnx/symbolic_caffe2.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import importlib
 import inspect
 

torch/onnx/symbolic_helper.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import functools

torch/onnx/symbolic_opset10.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import functools

torch/onnx/symbolic_opset11.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file exports ONNX ops for opset 11."""
 from __future__ import annotations
 

torch/onnx/symbolic_opset12.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 import functools

torch/onnx/symbolic_opset13.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # EDITING THIS FILE? READ THIS FIRST!
 # see Note [Edit Symbolic Files] in README.md
 

torch/onnx/symbolic_opset14.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file exports ONNX ops for opset 14.
 
 Note [ONNX operators that are added/updated in opset 14]

torch/onnx/symbolic_opset15.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file exports ONNX ops for opset 15.
 
 Note [ONNX operators that are added/updated in opset 15]

torch/onnx/symbolic_opset16.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file exports ONNX ops for opset 16.
 
 Note [ONNX Operators that are added/updated in opset 16]

torch/onnx/symbolic_opset17.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file exports ONNX ops for opset 17.
 
 Note [ONNX Operators that are added/updated in opset 17]

torch/onnx/symbolic_opset18.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """This file exports ONNX ops for opset 18.
 
 Note [ONNX Operators that are added/updated in opset 18]