Docstring doesn't show on hover in vscode jupyter notebook

Question

When I import a function into a vscode notebook, instead of getting a docstring I just get this:

Here is the function:

def balance_binary_target(df, target="", ratio=None):
    """
    Takes data frame X and target string
    Returns dataframe with upsampled balanced classes
    """
    classes = df[target].value_counts()
    class_differential = classes[0] - classes[1]
    if class_differential < 0:
        class_value = 0
        upsample = np.abs(class_differential)
        if ratio:
            upsample = int(ratio * upsample)
        sample = df[df[target] == class_value].sample(upsample, replace=True)
        df = pd.concat([sample, df])
    elif class_differential > 0:
        class_value = 1
        if ratio:
            upsample = int(ratio * upsample)
        upsample = np.abs(class_differential)
        sample = df[df[target] == class_value].sample(upsample, replace=True)
        df = pd.concat([sample, df])
    return df

When I hover over the same function in the simple file text editor I get this:

How can I get the same detail showing in the jupyter notebook as I do in the base editor?

Here is my version of vscode:

Here is my version of the Python extension: