Flip default value for mypy disallow_untyped_defs [6/11] (#127843)

See #127836 for details.

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

torch/distributed/elastic/utils/data/cycling_iterator.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

torch/distributed/elastic/utils/data/elastic_distributed_sampler.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

torch/distributed/elastic/utils/distributed.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

torch/distributed/elastic/utils/logging.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

torch/distributed/elastic/utils/store.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

torch/distributed/examples/memory_tracker_example.py

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

torch/distributed/fsdp/_common_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 This file includes private common utilities for FSDP.
 """

torch/distributed/fsdp/_debug_utils.py

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

torch/distributed/fsdp/_dynamo_utils.py

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

torch/distributed/fsdp/_exec_order_utils.py

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

torch/distributed/fsdp/_flat_param.py

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

torch/distributed/fsdp/_init_utils.py

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

torch/distributed/fsdp/_optim_utils.py

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

torch/distributed/fsdp/_runtime_utils.py

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

torch/distributed/fsdp/_shard_utils.py

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

torch/distributed/fsdp/_state_dict_utils.py

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

torch/distributed/fsdp/_trace_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import functools
 from contextlib import contextmanager
 from dataclasses import dataclass, field

torch/distributed/fsdp/_unshard_param_utils.py

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

torch/distributed/fsdp/_wrap_utils.py

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

torch/distributed/fsdp/sharded_grad_scaler.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import logging
 from collections import abc, defaultdict
 from typing import Any, Dict, Iterable, List, Optional, overload, Sequence, Tuple, Union

torch/distributed/fsdp/wrap.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Facebook, Inc. and its affiliates.
 #
 # This source code is licensed under the BSD license found in the

torch/distributed/launch.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""
 Module ``torch.distributed.launch``.
 

torch/distributed/launcher/api.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

torch/distributed/nn/api/remote_module.py

@@ -1,4 +1,5 @@
 #!/usr/bin/python3
+# mypy: allow-untyped-defs
 import collections
 import io
 import sys

torch/distributed/nn/functional.py

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

torch/distributed/nn/jit/instantiator.py

@@ -1,4 +1,5 @@
 #!/usr/bin/python3
+# mypy: allow-untyped-defs
 import importlib
 import logging
 import os

torch/distributed/nn/jit/templates/remote_module_template.py

@@ -1,4 +1,5 @@
 #!/usr/bin/python3
+# mypy: allow-untyped-defs
 
 
 def get_remote_module_template(enable_moving_cpu_tensors_to_cuda: bool):

torch/distributed/optim/functional_adadelta.py

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

torch/distributed/optim/functional_adagrad.py

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

torch/distributed/optim/functional_adam.py

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

torch/distributed/optim/functional_adamax.py

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

torch/distributed/optim/functional_adamw.py

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

torch/distributed/optim/functional_rmsprop.py

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

torch/distributed/optim/functional_rprop.py

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

torch/distributed/optim/functional_sgd.py

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

torch/distributed/optim/named_optimizer.py

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

torch/distributed/optim/optimizer.py

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

torch/distributed/optim/post_localSGD_optimizer.py

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

torch/distributed/optim/utils.py

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

torch/distributed/optim/zero_redundancy_optimizer.pyi

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

torch/distributed/pipelining/_IR.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 import copy
 import logging

torch/distributed/pipelining/_backward.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 from typing import List, Optional
 

torch/distributed/pipelining/_debug.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 import torch
 

torch/distributed/pipelining/_unflatten.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 from typing import Dict
 

torch/distributed/pipelining/_utils.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 import logging
 from dataclasses import dataclass

torch/distributed/pipelining/microbatch.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 import logging
 from typing import Any, Dict, List, Optional, Tuple

torch/distributed/pipelining/schedules.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 
 import logging

torch/distributed/pipelining/stage.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 import logging
 import operator

torch/distributed/remote_device.py

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

torch/distributed/rendezvous.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 try:
     from urllib.parse import urlparse, urlunparse
 except ImportError as e:

torch/distributed/rpc/__init__.py

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

torch/distributed/rpc/_testing/__init__.py

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

torch/distributed/rpc/_testing/faulty_agent_backend_registry.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 import torch.distributed as dist
 import torch.distributed.rpc as rpc

torch/distributed/rpc/_utils.py

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

torch/distributed/rpc/api.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 __all__ = ["shutdown", "get_worker_info", "remote", "rpc_sync",
            "rpc_async", "RRef", "AllGatherStates", "method_factory", "new_method"]
 

torch/distributed/rpc/backend_registry.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 __all__ = ["init_backend", "backend_registered", "construct_rpc_backend_options", "register_backend", "BackendType", "BackendValue"]
 
 import collections

torch/distributed/rpc/functions.py

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

torch/distributed/rpc/internal.py

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

torch/distributed/rpc/options.py

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

torch/distributed/rpc/rref_proxy.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from functools import partial
 
 from . import functions

torch/distributed/rpc/server_process_global_profiler.py

@@ -1,4 +1,5 @@
 #!/usr/bin/python3
+# mypy: allow-untyped-defs
 
 import itertools
 

torch/distributed/run.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# mypy: allow-untyped-defs
 
 # Copyright (c) Facebook, Inc. and its affiliates.
 # All rights reserved.

torch/distributed/tensor/parallel/_utils.py

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

torch/distributed/tensor/parallel/ddp.py

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

torch/distributed/tensor/parallel/fsdp.py

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

torch/distributed/tensor/parallel/loss.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 import contextlib
 from typing import cast, Dict, Optional, Tuple

torch/distributed/tensor/parallel/style.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 # Copyright (c) Meta Platforms, Inc. and affiliates
 from abc import ABC, abstractmethod
 from typing import Optional, Union, Tuple, Dict

torch/distributed/utils.py

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

torch/distributions/bernoulli.py

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

torch/distributions/beta.py

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

torch/distributions/binomial.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.distributions import constraints
 from torch.distributions.distribution import Distribution

torch/distributions/categorical.py

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

torch/distributions/cauchy.py

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

torch/distributions/chi2.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.distributions import constraints
 from torch.distributions.gamma import Gamma
 

torch/distributions/constraint_registry.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""
 PyTorch provides two global :class:`ConstraintRegistry` objects that link
 :class:`~torch.distributions.constraints.Constraint` objects to

torch/distributions/constraints.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 r"""
 The following constraints are implemented:
 

torch/distributions/continuous_bernoulli.py

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

torch/distributions/dirichlet.py

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

torch/distributions/distribution.py

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

torch/distributions/exp_family.py

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

torch/distributions/exponential.py

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

torch/distributions/fishersnedecor.py

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

torch/distributions/gamma.py

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

torch/distributions/geometric.py

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

torch/distributions/gumbel.py

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

torch/distributions/half_cauchy.py

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

torch/distributions/half_normal.py

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

torch/distributions/independent.py

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

torch/distributions/inverse_gamma.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch.distributions import constraints
 from torch.distributions.gamma import Gamma

torch/distributions/kl.py

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

torch/distributions/kumaraswamy.py

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

torch/distributions/laplace.py

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

torch/distributions/lkj_cholesky.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 """
 This closely follows the implementation in NumPyro (https://github.com/pyro-ppl/numpyro).
 

torch/distributions/log_normal.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.distributions import constraints
 from torch.distributions.normal import Normal
 from torch.distributions.transformed_distribution import TransformedDistribution

torch/distributions/logistic_normal.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 from torch.distributions import constraints
 from torch.distributions.normal import Normal
 from torch.distributions.transformed_distribution import TransformedDistribution

torch/distributions/lowrank_multivariate_normal.py

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

torch/distributions/mixture_same_family.py

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

torch/distributions/multinomial.py

@@ -1,3 +1,4 @@
+# mypy: allow-untyped-defs
 import torch
 from torch import inf
 from torch.distributions import Categorical, constraints

torch/distributions/multivariate_normal.py

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

torch/distributions/negative_binomial.py

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