Flip default value for mypy disallow_untyped_defs [4/11] (#127841)

See #127836 for details.

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

torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/data_sparsity.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import defaultdict
 from copy import deepcopy
 from typing import Any, Optional, Dict, TYPE_CHECKING

torch/ao/pruning/_experimental/data_sparsifier/lightning/tests/test_callbacks.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.ao.pruning._experimental.data_sparsifier.data_norm_sparsifier import DataNormSparsifier
 from torch.ao.pruning._experimental.data_scheduler.base_data_scheduler import BaseDataScheduler
 import torch

torch/ao/pruning/_experimental/data_sparsifier/quantization_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 import torch.nn as nn
 from torch.ao.pruning.sparsifier.utils import module_to_fqn, fqn_to_module

torch/ao/pruning/_experimental/pruner/FPGM_pruner.py

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

torch/ao/pruning/_experimental/pruner/base_structured_sparsifier.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from itertools import chain
 from operator import getitem
 import torch

torch/ao/pruning/_experimental/pruner/lstm_saliency_pruner.py

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

torch/ao/pruning/_experimental/pruner/parametrization.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch import nn
 from torch.nn.utils.parametrize import is_parametrized

torch/ao/pruning/_experimental/pruner/prune_functions.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 Collection of conversion functions for linear / conv2d structured pruning
 Also contains utilities for bias propagation

torch/ao/pruning/_experimental/pruner/saliency_pruner.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from .base_structured_sparsifier import BaseStructuredSparsifier
 
 

torch/ao/pruning/_mappings.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 __all__ = [
     "get_static_sparse_quantized_mapping",
     "get_dynamic_sparse_quantized_mapping",

torch/ao/pruning/scheduler/base_scheduler.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 from torch.ao.pruning import BaseSparsifier
 

torch/ao/pruning/scheduler/cubic_scheduler.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import warnings
 
 from .base_scheduler import BaseScheduler

torch/ao/pruning/scheduler/lambda_scheduler.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import warnings
 
 from .base_scheduler import BaseScheduler

torch/ao/pruning/sparsifier/base_sparsifier.py

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

torch/ao/pruning/sparsifier/nearly_diagonal_sparsifier.py

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

torch/ao/pruning/sparsifier/utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import Any, Dict, Optional, Type
 from torch.nn.utils.parametrize import type_before_parametrizations, is_parametrized
 from itertools import chain

torch/ao/pruning/sparsifier/weight_norm_sparsifier.py

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

torch/ao/quantization/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # flake8: noqa: F403
 
 from .fake_quantize import *  # noqa: F403

torch/ao/quantization/_correct_bias.py

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

torch/ao/quantization/_equalize.py

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

torch/ao/quantization/_learnable_fake_quantize.py

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

torch/ao/quantization/backend_config/_common_operator_config_utils.py

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

torch/ao/quantization/backend_config/_qnnpack_pt2e.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import operator
 import torch
 from torch.ao.quantization.backend_config import (

torch/ao/quantization/backend_config/backend_config.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from __future__ import annotations
 from dataclasses import dataclass
 from typing import Any, Callable, Dict, List, Optional, Type, Union, TYPE_CHECKING

torch/ao/quantization/backend_config/native.py

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

torch/ao/quantization/backend_config/onednn.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/quantization/backend_config/tensorrt.py

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

torch/ao/quantization/backend_config/utils.py

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

torch/ao/quantization/experimental/APoT_tensor.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.ao.quantization.experimental.quantizer import APoTQuantizer
 

torch/ao/quantization/experimental/adaround_fake_quantize.py

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

torch/ao/quantization/experimental/adaround_optimization.py

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

torch/ao/quantization/experimental/apot_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""
 This file contains utility functions to convert values
 using APoT nonuniform quantization methods.

torch/ao/quantization/experimental/fake_quantize.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch import Tensor
 from torch.ao.quantization.experimental.observer import APoTObserver

torch/ao/quantization/experimental/fake_quantize_function.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch import Tensor
 from torch.ao.quantization.experimental.quantizer import quantize_APoT, dequantize_APoT

torch/ao/quantization/experimental/linear.py

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

torch/ao/quantization/experimental/observer.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 This module implements nonuniform observers used to collect statistics about
 the values observed during calibration (PTQ) or training (QAT).

torch/ao/quantization/experimental/quantizer.py

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

torch/ao/quantization/fake_quantize.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """Implements modules  used to perform fake quantization."""
 
 import torch

torch/ao/quantization/fuse_modules.py

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

torch/ao/quantization/fuser_method_mappings.py

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

torch/ao/quantization/fx/_decomposed.py

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

torch/ao/quantization/fx/_equalize.py

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

torch/ao/quantization/fx/_lower_to_native_backend.py

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

torch/ao/quantization/fx/_model_report/detector.py

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

torch/ao/quantization/fx/_model_report/model_report.py

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

torch/ao/quantization/fx/_model_report/model_report_observer.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.ao.quantization.observer import ObserverBase
 

torch/ao/quantization/fx/_model_report/model_report_visualizer.py

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

torch/ao/quantization/fx/custom_config.py

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

torch/ao/quantization/fx/fuse.py

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

torch/ao/quantization/fx/fuse_handler.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.ao.quantization.backend_config import BackendConfig
 from torch.fx.graph import Node, Graph

torch/ao/quantization/fx/graph_module.py

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

torch/ao/quantization/fx/match_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import sys
 import torch
 from torch.fx.graph import (

torch/ao/quantization/fx/pattern_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import OrderedDict
 from typing import Dict, Any
 from torch.ao.quantization.utils import Pattern

torch/ao/quantization/fx/prepare.py

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

torch/ao/quantization/fx/qconfig_mapping_utils.py

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

torch/ao/quantization/fx/quantize_handler.py

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

torch/ao/quantization/fx/utils.py

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

torch/ao/quantization/observer.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 This module implements observers which are used to collect statistics about
 the values observed during calibration (PTQ) or training (QAT).

torch/ao/quantization/pt2e/duplicate_dq_pass.py

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

torch/ao/quantization/pt2e/export_utils.py

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

torch/ao/quantization/pt2e/graph_utils.py

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

torch/ao/quantization/pt2e/port_metadata_pass.py

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

torch/ao/quantization/pt2e/prepare.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch._subclasses import FakeTensor
 from torch.ao.quantization.fx.prepare import (

torch/ao/quantization/pt2e/qat_utils.py

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

torch/ao/quantization/pt2e/representation/rewrite.py

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

torch/ao/quantization/pt2e/utils.py

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

torch/ao/quantization/qconfig.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from collections import namedtuple
 from typing import Optional, Any, Union, Type
 from typing_extensions import deprecated

torch/ao/quantization/qconfig_mapping.py

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

torch/ao/quantization/quantize.py

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

torch/ao/quantization/quantize_jit.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 
 import torch
 from torch.ao.quantization.qconfig import QConfig

torch/ao/quantization/quantizer/embedding_quantizer.py

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

torch/ao/quantization/quantizer/quantizer.py

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

torch/ao/quantization/quantizer/utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from typing import List
 
 from torch.ao.quantization.pt2e.utils import _is_sym_size_node

torch/ao/quantization/quantizer/x86_inductor_quantizer.py

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

torch/ao/quantization/quantizer/xnnpack_quantizer.py

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

torch/ao/quantization/quantizer/xnnpack_quantizer_utils.py

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

torch/ao/quantization/stubs.py

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

torch/ao/quantization/utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 Utils shared by different modes of quantization (eager/graph)
 """

torch/autograd/__init__.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 ``torch.autograd`` provides classes and functions implementing automatic
 differentiation of arbitrary scalar valued functions. It requires minimal

torch/autograd/_functions/tensor.py

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

torch/autograd/_functions/utils.py

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

torch/autograd/anomaly_mode.py

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

torch/autograd/forward_ad.py

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

torch/autograd/function.py

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

torch/autograd/functional.py

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

torch/autograd/grad_mode.py

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

torch/autograd/gradcheck.py

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

torch/autograd/graph.py

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

torch/autograd/profiler.py

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

torch/autograd/profiler_legacy.py

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

torch/autograd/profiler_util.py

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

torch/autograd/variable.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch._C import _ImperativeEngine as ImperativeEngine
 

torch/backends/__init__.py

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

torch/backends/_coreml/preprocess.py

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

torch/backends/_nnapi/prepare.py

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

torch/backends/_nnapi/serializer.py

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

torch/backends/cuda/__init__.py

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

torch/backends/cudnn/__init__.py

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

torch/backends/cudnn/rnn.py

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

torch/backends/mkl/__init__.py

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