Flip default value for mypy disallow_untyped_defs [7/11] (#127844)

See #127836 for details.

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

torch/export/exported_program.py

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

torch/export/graph_signature.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import dataclasses
 from enum import auto, Enum
 from typing import Collection, Dict, List, Mapping, Optional, Set, Tuple, Union

torch/export/unflatten.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import abc
 import copy
 import operator

torch/functional.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import (
     List, Tuple, Optional, Union, Any, Sequence, TYPE_CHECKING
 )

torch/futures/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 
 from typing import cast, Callable, Generic, List, Optional, Type, TypeVar, Union

torch/fx/_compatibility.py

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

torch/fx/_lazy_graph_module.py

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

torch/fx/_pytree.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import namedtuple
 from typing import Any, Callable, Dict, List, NamedTuple, Optional, Tuple, Type
 

torch/fx/_symbolic_trace.py

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

torch/fx/_utils.py

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

torch/fx/annotate.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.fx.proxy import Proxy
 from ._compatibility import compatibility
 

torch/fx/experimental/_sym_dispatch_mode.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import List, Optional, Type
 
 __all__ = ["SymDispatchMode", "handle_sym_dispatch", "sym_function_mode"]

torch/fx/experimental/accelerator_partitioner.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import operator
 from collections import deque
 from typing import Dict, List, Set, NamedTuple, Tuple, Deque

torch/fx/experimental/const_fold.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import re
 from typing import Callable, Dict, Optional, Set, Union
 

torch/fx/experimental/debug.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch.fx as fx
 
 def set_trace(gm: fx.GraphModule) -> fx.GraphModule:

torch/fx/experimental/graph_gradual_typechecker.py

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

torch/fx/experimental/merge_matmul.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch.fx.node import Node

torch/fx/experimental/meta_tracer.py

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

torch/fx/experimental/migrate_gradual_types/constraint.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.fx.experimental.migrate_gradual_types.operation import op_add, op_sub, op_mul, op_div, \
     op_mod, op_gt, op_lt, op_neq, op_eq
 from torch.fx.tensor_type import TensorType, Dyn

torch/fx/experimental/migrate_gradual_types/constraint_generator.py

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

torch/fx/experimental/migrate_gradual_types/transform_to_z3.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.fx.experimental.migrate_gradual_types.constraint import Conj, Disj, T, F, BinConstraintT, BVar, is_bool_expr
 from torch.fx.experimental.migrate_gradual_types.constraint import BinConstraintD, TVar, DVar
 from torch.fx.experimental.migrate_gradual_types.constraint import Prod, is_algebraic_expression, is_dim

torch/fx/experimental/migrate_gradual_types/util.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.fx.experimental.migrate_gradual_types.constraint import TVar, DVar, BinConstraintD, \
     BVar
 from torch.fx.experimental.migrate_gradual_types.operation import op_leq

torch/fx/experimental/normalize.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import operator
 from typing import Any, Callable, Dict, Tuple, Optional
 

torch/fx/experimental/optimization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch.fx as fx
 from torch.fx.node import Argument, Target
 from torch.nn.utils.fusion import fuse_conv_bn_eval

torch/fx/experimental/partitioner_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from enum import Enum
 from typing import NamedTuple, Dict, List, Set
 

torch/fx/experimental/recording.py

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

torch/fx/experimental/refinement_types.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 class Equality:
     def __init__(self, lhs, rhs):
         self.lhs = lhs

torch/fx/experimental/rewriter.py

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

torch/fx/experimental/schema_type_annotation.py

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

torch/fx/experimental/shape_inference/infer_shape.py

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

torch/fx/experimental/sym_node.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 This file does three things:
 - Contains the definition of SymNode

torch/fx/experimental/unification/core.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections.abc import Iterator  # type: ignore[import]
 from functools import partial
 

torch/fx/experimental/unification/match.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .core import unify, reify  # type: ignore[attr-defined]
 from .variable import isvar
 from .utils import _toposort, freeze

torch/fx/experimental/unification/more.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .core import unify, reify  # type: ignore[attr-defined]
 from .dispatch import dispatch
 

torch/fx/experimental/unification/multipledispatch/conflict.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .utils import _toposort, groupby
 from .variadic import isvariadic
 import operator

torch/fx/experimental/unification/multipledispatch/core.py

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

torch/fx/experimental/unification/multipledispatch/dispatcher.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from warnings import warn
 import inspect
 from typing_extensions import deprecated

torch/fx/experimental/unification/multipledispatch/utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import OrderedDict
 
 __all__ = ["raises", "expand_tuples", "reverse_dict", "groupby", "typename"]

torch/fx/experimental/unification/multipledispatch/variadic.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .utils import typename
 
 __all__ = ["VariadicSignatureType", "isvariadic", "VariadicSignatureMeta", "Variadic"]

torch/fx/experimental/unification/unification_tools.py

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

torch/fx/experimental/unification/utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 __all__ = ["hashable", "transitive_get", "raises", "reverse_dict", "xfail", "freeze"]
 def hashable(x):
     try:

torch/fx/experimental/unification/variable.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from contextlib import contextmanager
 from .utils import hashable
 from .dispatch import dispatch

torch/fx/experimental/unify_refinements.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.fx.experimental.graph_gradual_typechecker import Refine
 from torch.fx.tensor_type import TensorType
 from torch.fx.experimental.unification import Var, unify  # type: ignore[attr-defined]

torch/fx/experimental/validator.py

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

torch/fx/graph.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import defaultdict
 from .node import Node, Argument, Target, map_arg, _type_repr, _get_qualified_name
 import torch.utils._pytree as pytree

torch/fx/graph_module.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import contextlib
 import copy
 import itertools

torch/fx/immutable_collections.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Any, Dict, Iterable, List, Tuple
 
 from torch.utils._pytree import (

torch/fx/interpreter.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .graph_module import GraphModule
 from ._lazy_graph_module import _make_graph_module
 from .graph import Graph

torch/fx/operator_schemas.py

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

torch/fx/passes/backends/cudagraphs.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.fx.passes.infra.partitioner import CapabilityBasedPartitioner
 from torch.fx.passes.operator_support import OperatorSupport

torch/fx/passes/dialect/common/cse_pass.py

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

torch/fx/passes/fake_tensor_prop.py

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

torch/fx/passes/graph_drawer.py

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

torch/fx/passes/graph_manipulation.py

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

torch/fx/passes/graph_transform_observer.py

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

torch/fx/passes/infra/partitioner.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.fx.passes.utils.fuser_utils import fuse_by_partitions
 import collections
 import itertools

torch/fx/passes/infra/pass_base.py

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

torch/fx/passes/infra/pass_manager.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import inspect
 import logging
 from queue import Queue

torch/fx/passes/net_min_base.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import logging
 from dataclasses import dataclass
 from typing import Any, Callable, Dict, List, Optional, Tuple

torch/fx/passes/operator_support.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import abc
 import typing as t
 

torch/fx/passes/pass_manager.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from functools import wraps
 from inspect import unwrap
 from typing import Callable, List, Optional

torch/fx/passes/reinplace.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.fx import Node
 from torch.fx._compatibility import compatibility

torch/fx/passes/runtime_assert.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import logging
 import operator
 from typing import Any, Dict, Optional, Set, TYPE_CHECKING

torch/fx/passes/split_module.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import inspect
 from typing import Any, Callable, Dict, List, Optional, Set
 from collections import OrderedDict

torch/fx/passes/split_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import copy
 from dataclasses import dataclass, field
 from typing import Dict, List, Optional, Tuple, Type, Union

torch/fx/passes/splitter_base.py

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

torch/fx/passes/tools_common.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import List, Tuple, Union, Dict, Any, Set, Mapping, Optional
 import collections
 from dataclasses import dataclass

torch/fx/passes/utils/common.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Dict, Tuple
 
 from torch.fx._compatibility import compatibility

torch/fx/passes/utils/fuser_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import copy
 from queue import SimpleQueue
 from typing import List, Dict, Tuple

torch/fx/passes/utils/matcher_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from dataclasses import dataclass, field
 from collections import defaultdict
 import copy

torch/fx/passes/utils/source_matcher_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from dataclasses import dataclass, field
 from torch.fx.graph import Graph
 from torch.fx.node import Node

torch/fx/tensor_type.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.fx.experimental.unification import Var  # type: ignore[attr-defined]
 
 from ._compatibility import compatibility

torch/fx/traceback.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import traceback
 from contextlib import contextmanager
 from typing import List, Any, Dict

torch/hub.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import contextlib
 import errno
 import hashlib

torch/jit/__init__.py

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

torch/jit/_async.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Async API.
 
 This module contains the API for parallelism in TorchScript, notably:

torch/jit/_await.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch._jit_internal import _Await
 from torch.jit._builtins import _register_builtin

torch/jit/_builtins.py

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

torch/jit/_check.py

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

torch/jit/_dataclass_impls.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Functions for synthesizing magic methods for JIT-compiled dataclasses
 import ast
 import dataclasses

torch/jit/_decomposition_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch._ops import OpOverload, OpOverloadPacket
 

torch/jit/_decompositions.py

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

torch/jit/_freeze.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Freezing.
 
 This is not intended to be imported directly; please use the exposed

torch/jit/_fuser.py

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

torch/jit/_ir_utils.py

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

torch/jit/_monkeytype_config.py

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

torch/jit/_passes/_property_propagation.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 Tools to help with tensor property propagation.
 

torch/jit/_pickle.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # These functions are referenced from the pickle archives produced by
 # ScriptModule.save()
 

torch/jit/_recursive.py

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

torch/jit/_script.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # mypy: disable-error-code="type-arg"
 from typing import (
     Any,

torch/jit/_serialization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Serialization.
 
 This module contains functionality for serializing TorchScript modules, notably:

torch/jit/_shape_functions.py

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

torch/jit/_state.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """JIT-related state.
 
 This module stores various pieces of Python-global state relating to the JIT.

torch/jit/_trace.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Tracing.
 
 This module contains functionality to support the JIT's tracing frontend, notably:

torch/jit/annotations.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import ast
 import builtins
 import dis

torch/jit/frontend.py

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

torch/jit/generate_bytecode.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import List
 
 from torch._C import _compile_graph_to_code_table, _generate_upgraders_graph

torch/jit/mobile/__init__.py

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

torch/jit/quantized.py

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

torch/jit/supported_ops.py

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