Flip default value for mypy disallow_untyped_defs [1/11] (#127838)

See #127836 for details.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/127838
Approved by: https://github.com/oulgen

caffe2/perfkernels/hp_emblookup_codegen.py

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

test/test_bundled_images.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # Owner(s): ["oncall: mobile"]
+# mypy: allow-untyped-defs
 
 import io
 

test/test_bundled_inputs.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # Owner(s): ["oncall: mobile"]
+# mypy: allow-untyped-defs
 
 import io
 import textwrap

test/test_complex.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Owner(s): ["module: complex"]
 
 import torch

test/test_futures.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Owner(s): ["module: unknown"]
 
 import threading

test/test_torch.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Owner(s): ["module: tests"]
 
 import torch

test/test_type_hints.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Owner(s): ["module: typing"]
 
 import doctest

test/test_type_info.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Owner(s): ["module: typing"]
 
 from torch.testing._internal.common_utils import (

test/test_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Owner(s): ["module: unknown"]
 
 import os

torch/_C/_VariableFunctions.pyi.in

@@ -1,5 +1,6 @@
 # ${generated_comment}
 # mypy: disable-error-code="type-arg"
+# mypy: allow-untyped-defs
 
 import builtins
 from typing import (

torch/_C/__init__.pyi.in

@@ -1,5 +1,6 @@
 # ${generated_comment}
 # mypy: disable-error-code="type-arg"
+# mypy: allow-untyped-defs
 
 import builtins
 from enum import Enum, IntEnum

torch/_C/_autograd.pyi

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

torch/_C/_distributed_autograd.pyi

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

torch/_C/_distributed_c10d.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # mypy: disable-error-code="type-arg"
 from datetime import timedelta
 from enum import Enum

torch/_C/_distributed_rpc.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # mypy: disable-error-code="type-arg"
 from datetime import timedelta
 from typing import Any, Dict, Generic, List, Optional, overload, Tuple, Type, TypeVar

torch/_C/_dynamo/eval_frame.pyi

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

torch/_C/_dynamo/guards.pyi

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

torch/_C/_functorch.pyi

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

torch/_C/_lazy.pyi

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

torch/_C/_lazy_ts_backend.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # defined in torch/csrc/lazy/python/init.cpp
 
 from typing import Any, List, Tuple

torch/_C/_nvtx.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Defined in torch/csrc/cuda/shared/nvtx.cpp
 def rangePushA(message: str) -> int: ...
 def rangePop() -> int: ...

torch/_C_flatbuffer/__init__.pyi

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch._C import LiteScriptModule, ScriptModule
 
 def _load_mobile_module_from_file(filename: str): ...

torch/__config__.py

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

torch/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 r"""
 The torch package contains data structures for multi-dimensional

torch/_classes.py

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

torch/_compile.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 APIs related to torch.compile which lazily import torch._dynamo to avoid
 circular dependencies.

torch/_custom_op/autograd.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.utils._pytree as pytree
 from collections import namedtuple

torch/_custom_op/functional.py

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

torch/_custom_op/impl.py

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

torch/_custom_ops.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import inspect
 
 from torch._custom_op.impl import (

torch/_decomp/__init__.py

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

torch/_decomp/decompositions.py

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

torch/_decomp/decompositions_for_jvp.py

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

torch/_decomp/decompositions_for_rng.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import functools
 from collections import defaultdict
 from typing import Callable, Dict

torch/_deploy.py

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

torch/_dispatch/python.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import itertools
 import unittest.mock
 from contextlib import contextmanager

torch/_dynamo/_trace_wrapped_higher_order_op.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch._C import DispatchKey
 from torch._higher_order_ops.utils import autograd_not_implemented

torch/_dynamo/bytecode_analysis.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import bisect
 import dataclasses
 import dis

torch/_dynamo/bytecode_transformation.py

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

torch/_dynamo/cache_size.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import logging
 import types
 import weakref

torch/_dynamo/callback.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 class CompilationCallbackHandler:
     def __init__(self):
         self.start_callbacks = []

torch/_dynamo/code_context.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import types
 
 from .utils import ExactWeakKeyDictionary

torch/_dynamo/codegen.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import collections
 import dataclasses
 import re

torch/_dynamo/compiled_autograd.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import contextlib
 import functools
 from typing import Dict, List, Optional, TYPE_CHECKING

torch/_dynamo/comptime.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # This file establishes the public comptime interface to Dynamo.
 # This allows Dynamo users to execute arbitrary Python code while
 # Dynamo is symbolically evaluating their original programs.

torch/_dynamo/config.py

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

torch/_dynamo/convert_frame.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import collections
 import cProfile
 import dis

torch/_dynamo/create_parameter_op.py

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

torch/_dynamo/current_scope_id.py

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

torch/_dynamo/debug_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # mypy: disable-error-code="method-assign"
 
 import copy

torch/_dynamo/decorators.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # ruff: noqa: TCH004
 from dataclasses import dataclass
 from typing import TYPE_CHECKING

torch/_dynamo/device_interface.py

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

torch/_dynamo/eval_frame.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # mypy: disable-error-code="method-assign"
 
 """

torch/_dynamo/exc.py

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

torch/_dynamo/external_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # This module contains functions that *will be allowed* by dynamo
 
 import functools

torch/_dynamo/guards.py

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

torch/_dynamo/logging.py

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

torch/_dynamo/mutation_guard.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # mypy: disable-error-code="method-assign"
 
 import functools

torch/_dynamo/output_graph.py

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

torch/_dynamo/profiler.py

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

torch/_dynamo/replay_record.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import dataclasses
 from dataclasses import field
 from types import CodeType, ModuleType

torch/_dynamo/repro/after_aot.py

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

torch/_dynamo/repro/after_dynamo.py

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

torch/_dynamo/resume_execution.py

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

torch/_dynamo/side_effects.py

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

torch/_dynamo/source.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import collections
 import dataclasses
 import enum

torch/_dynamo/symbolic_convert.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import collections
 import collections.abc
 import contextlib

torch/_dynamo/tensor_version_op.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch._prims import _make_prim, RETURN_TYPE
 from torch._subclasses import FakeTensorMode

torch/_dynamo/test_case.py

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

torch/_dynamo/test_minifier_common.py

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

torch/_dynamo/testing.py

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

torch/_dynamo/trace_rules.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import _collections_abc
 import _weakrefset
 import abc

torch/_dynamo/utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import atexit
 import collections
 import contextlib

torch/_dynamo/variables/script_object.py

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

torch/_dynamo/variables/torch.py

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

torch/_export/__init__.py

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

torch/_export/converter.py

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

torch/_export/db/case.py

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

torch/_export/db/examples/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import glob
 import importlib
 from os.path import basename, dirname, isfile, join

torch/_export/db/examples/assume_constant_result.py

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

torch/_export/db/examples/autograd_function.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/class_method.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/cond_branch_class_method.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/cond_branch_nested_function.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/cond_branch_nonlocal_variables.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/cond_closed_over_variable.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/cond_operands.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/cond_predicate.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/constrain_as_size_example.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/constrain_as_value_example.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/decorator.py

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

torch/_export/db/examples/dictionary.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/dynamic_shape_assert.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/dynamic_shape_constructor.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/dynamic_shape_if_guard.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/dynamic_shape_map.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/dynamic_shape_round.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case, SupportLevel

torch/_export/db/examples/dynamic_shape_slicing.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/dynamic_shape_view.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case

torch/_export/db/examples/fn_with_kwargs.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._export.db.case import export_case, ExportArgs, SupportLevel