Argument suggestions for some functions in the pytorch module are duplicated when using Jedi

Reproduce

import time
import torch
from torch import nn, optim
import sys
import d2lzh_pytorch as d2l

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')


def nin_block(in_channels, out_channels, kernel_size, stride, padding):
    blk = nn.Sequential(
        nn.Conv2d(in_channels, out_channels, kernel_size, stride, pad)
    )

[CatNofishing]

@ericsnowcurrently @karthiknadig This problem still exists, do you have any plans to fix this problem, otherwise, I can only use pylance as an alternative.

[CatNofishing]

Okay, it works better than jedi , but worse than pylance😥