pytorch - 💡(How to fix) Fix ModuleList not recognized as Iterable by type checkers [1 pull requests]

import torch
from torch import nn

class MyModel(nn.Module):
    def __init__(self) -> None:
        super().__init__()
        self.blocks = nn.ModuleList([nn.Linear(4, 4) for _ in range(3)])
        self.norms = nn.ModuleList([nn.LayerNorm(4) for _ in range(3)])

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        for block in self.blocks:  # ty: not-iterable
            x = block(x)
        return x

    def paired(self, x: torch.Tensor) -> torch.Tensor:
        for block, norm in zip(self.blocks, self.norms):  # ty: not-iterable
            x = norm(block(x))
        return x