Flip default value for mypy disallow_untyped_defs [3/11] (#127840)

See #127836 for details.

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

torch/_inductor/remote_cache.py

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

torch/_inductor/runtime/compile_tasks.py

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

torch/_inductor/runtime/coordinate_descent_tuner.py

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

torch/_inductor/runtime/hints.py

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

torch/_inductor/runtime/runtime_utils.py

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

torch/_inductor/runtime/triton_helpers.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 try:
     import triton
     import triton.language as tl

torch/_inductor/runtime/triton_heuristics.py

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

torch/_inductor/select_algorithm.py

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

torch/_inductor/sizevars.py

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

torch/_inductor/subgraph_lowering.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Utilities for lowering subgraphs used by higher order operators
 
 """

torch/_inductor/test_case.py

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

torch/_inductor/test_operators.py

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

torch/_inductor/utils.py

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

torch/_inductor/virtualized.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 This file provides a number of "global" variables/handlers that are actually
 thread local and dynamically scoped, with Inductor patching them to various

torch/_inductor/wrapper_benchmark.py

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

torch/_jit_internal.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 The weak_script annotation needs to be here instead of inside torch/jit/ so it
 can be used in other places in torch/ (namely torch.nn) without running into

torch/_lazy/__init__.py

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

torch/_lazy/closure.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import os
 import threading
 from queue import Empty as EmptyQueue, Queue

torch/_lazy/computation.py

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

torch/_lazy/config.py

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

torch/_lazy/debug.py

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

torch/_lazy/device_context.py

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

torch/_lazy/extract_compiled_graph.py

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

torch/_lazy/ir_cache.py

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

torch/_lazy/metrics.py

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

torch/_lazy/ts_backend.py

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

torch/_library/abstract_impl.py

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

torch/_library/autograd.py

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

torch/_library/custom_ops.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import inspect
 import weakref
 from typing import (

torch/_library/fake_class_registry.py

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

torch/_library/infer_schema.py

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

torch/_library/simple_registry.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .abstract_impl import AbstractImplHolder
 
 __all__ = ["SimpleLibraryRegistry", "SimpleOperatorEntry", "singleton"]

torch/_library/utils.py

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

torch/_linalg_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Various linear algebra utility methods for internal use.
 
 """

torch/_lobpcg.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Locally Optimal Block Preconditioned Conjugate Gradient methods.
 """
 # Author: Pearu Peterson

torch/_logging/_internal.py

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

torch/_meta_registrations.py

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

torch/_namedtensor_internals.py

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

torch/_ops.py

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

torch/_prims/__init__.py

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

torch/_prims/context.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import functools
 from contextlib import nullcontext
 from typing import Any, Callable, Dict, Optional, Sequence

torch/_prims/debug_prims.py

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

torch/_prims/executor.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Callable, Optional
 
 from torch._prims.context import TorchRefsMode

torch/_prims/rng_prims.py

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

torch/_prims_common/__init__.py

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

torch/_prims_common/wrappers.py

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

torch/_python_dispatcher.py

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

torch/_refs/__init__.py

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

torch/_refs/_conversions.py

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

torch/_refs/linalg/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from functools import partial
 
 from typing import List, Optional, Tuple, Union

torch/_refs/nn/functional/__init__.py

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

torch/_refs/special/__init__.py

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

torch/_size_docs.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Adds docstrings to torch.Size functions"""
 
 import torch._C

torch/_sources.py

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

torch/_storage_docs.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Adds docstrings to Storage functions"""
 
 import torch._C

torch/_streambase.py

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

torch/_strobelight/examples/cli_function_profiler_example.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._strobelight.cli_function_profiler import (

torch/_strobelight/examples/compile_time_profile_example.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 from torch._strobelight.compile_time_profiler import StrobelightCompileTimeProfiler

torch/_subclasses/fake_tensor.py

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

torch/_subclasses/functional_tensor.py

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

torch/_subclasses/meta_utils.py

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

torch/_tensor.py

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

torch/_tensor_docs.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Adds docstrings to Tensor functions"""
 
 import torch._C

torch/_tensor_str.py

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

torch/_torch_docs.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Adds docstrings to functions defined in the torch._C module."""
 
 import re

torch/_utils.py

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

torch/_utils_internal.py

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

torch/_vmap_internals.py

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

torch/_weights_only_unpickler.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Unpickler restricted to loading only state dicts
 # Restrict constructing types to a list defined in _get_allowed_globals()
 # Restrict BUILD operation to `Tensor`, `Parameter` and `OrderedDict` types only

torch/amp/autocast_mode.py

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

torch/amp/grad_scaler.py

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

torch/ao/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # torch.ao is a package with a lot of interdependencies.
 # We will use lazy import to avoid cyclic dependencies here.
 

torch/ao/nn/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # We are exposing all subpackages to the end-user.
 # Because of possible inter-dependency, we want to avoid
 # the cyclic imports, thus implementing lazy version

torch/ao/nn/intrinsic/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .modules import *  # noqa: F403
 from .modules.fused import _FusedModule  # noqa: F403
 

torch/ao/nn/intrinsic/modules/fused.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.nn import Conv1d, Conv2d, Conv3d, ReLU, Linear, BatchNorm1d, BatchNorm2d, BatchNorm3d
 from torch.nn.utils.parametrize import type_before_parametrizations

torch/ao/nn/intrinsic/qat/modules/conv_fused.py

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

torch/ao/nn/intrinsic/qat/modules/linear_fused.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn as nn
 import torch.ao.nn.intrinsic as nni

torch/ao/nn/intrinsic/qat/modules/linear_relu.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.ao.nn.qat as nnqat
 import torch.ao.nn.intrinsic as nni

torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.ao.nn.quantized.dynamic as nnqd
 import torch.ao.nn.intrinsic as nni

torch/ao/nn/intrinsic/quantized/modules/bn_relu.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 import torch
 import torch.ao.nn.intrinsic

torch/ao/nn/intrinsic/quantized/modules/conv_add.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.ao.nn.intrinsic
 import torch.ao.nn.intrinsic.qat

torch/ao/nn/intrinsic/quantized/modules/conv_relu.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 import torch
 import torch.ao.nn.intrinsic

torch/ao/nn/intrinsic/quantized/modules/linear_relu.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.ao.nn.quantized as nnq
 import torch.ao.nn.intrinsic as nni

torch/ao/nn/qat/dynamic/modules/linear.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 __all__ = ["Linear"]

torch/ao/nn/qat/modules/conv.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn as nn
 from torch.nn.modules.utils import _single, _pair, _triple

torch/ao/nn/qat/modules/embedding_ops.py

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

torch/ao/nn/qat/modules/linear.py

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

torch/ao/nn/quantizable/modules/activation.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.jit  # this is needed to avoid a circular import
 from torch import nn

torch/ao/nn/quantizable/modules/rnn.py

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

torch/ao/nn/quantized/dynamic/modules/conv.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""Dynamically quantized convolution modules."""
 
 import torch

torch/ao/nn/quantized/dynamic/modules/linear.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.ao.nn.quantized as nnq
 from torch.ao.nn.quantized.modules.utils import _quantize_weight

torch/ao/nn/quantized/dynamic/modules/rnn.py

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

torch/ao/nn/quantized/functional.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r""" Functional interface (quantized)."""
 from typing import List, Optional
 import warnings

torch/ao/nn/quantized/modules/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 # The quantized modules use `torch.nn` and `torch.ao.nn.quantizable`

torch/ao/nn/quantized/modules/activation.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from warnings import warn
 __all__ = [

torch/ao/nn/quantized/modules/batchnorm.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.ao.nn.intrinsic as nni
 

torch/ao/nn/quantized/modules/conv.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""Quantized convolution modules."""
 
 from typing import Optional, List, TypeVar

torch/ao/nn/quantized/modules/dropout.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 
 __all__ = ['Dropout']

torch/ao/nn/quantized/modules/embedding_ops.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn as nn
 from torch import Tensor  # noqa: F401

torch/ao/nn/quantized/modules/functional_modules.py

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