Skip to content

Commit 8dd3808

Browse files
bobrenjc93pytorchmergebot
authored andcommitted
remove allow-untyped-defs for torch/_functorch/batch_norm_replacement.py (pytorch#143438)
Pull Request resolved: pytorch#143438 Approved by: https://github.com/oulgen
1 parent 75fe5a3 commit 8dd3808

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

torch/_functorch/batch_norm_replacement.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,8 @@
1-
# mypy: allow-untyped-decorators
2-
# mypy: allow-untyped-defs
31
import torch.nn as nn
42
from torch._functorch.utils import exposed_in
53

64

7-
def batch_norm_without_running_stats(module: nn.Module):
5+
def batch_norm_without_running_stats(module: nn.Module) -> None:
86
if (
97
isinstance(module, nn.modules.batchnorm._BatchNorm)
108
and module.track_running_stats

0 commit comments

Comments
 (0)