如何在python docstring中指定不同的返回类型

def add(a: int, b: int) -> int:
    """
    This function takes two integers as input and returns their sum.

    Parameters:
        a (int): The first integer.
        b (int): The second integer.

    Returns:
        int: The sum of the two integers.
    """
    return a + b

from typing import Union

def divide(a: int, b: int) -> Union[int, float]:
    """
    This function takes two integers as input and returns their division.
    If the division is exact, it returns an integer; otherwise, it returns a float.

    Parameters:
        a (int): The dividend.
        b (int): The divisor.

    Returns:
        Union[int, float]: The result of the division.
    """
    if a % b == 0:
        return a // b
    else:
        return a / b