[Typing][B-18] Add type annotations for python/paddle/distribution/laplace.py

[ooooo-create]

PR Category

User Experience

PR Types

Improvements

Description

类型标注：
- python/paddle/distribution/laplace.py

Related links

• [Type Hints] 为公开 API 标注类型提示信息 #65008

@SigureMo @megemini

[paddle-bot]

你的PR提交成功，感谢你对开源项目的贡献!
请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[megemini]

用 float | Tensor ？

Numberic: TypeAlias = Union[int, float, complex, np.number, "Tensor"] 中的 complex 貌似不能支持？

In [4]:             >>> import paddle
   ...:             >>> paddle.seed(2023)
   ...:             >>> m = paddle.distribution.Laplace(1+1j, 2+1j)
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[4], line 3
      1 import paddle
      2 paddle.seed(2023)
----> 3 m = paddle.distribution.Laplace(1+1j, 2+1j)

File ~/venv38dev/lib/python3.8/site-packages/paddle/distribution/laplace.py:60, in Laplace.__init__(self, loc, scale)
     56 def __init__(self, loc, scale):
     57     if not isinstance(
     58         loc, (numbers.Real, framework.Variable, paddle.pir.Value)
     59     ):
---> 60         raise TypeError(
     61             f"Expected type of loc is Real|Variable, but got {type(loc)}"
     62         )
     64     if not isinstance(
     65         scale, (numbers.Real, framework.Variable, paddle.pir.Value)
     66     ):
     67         raise TypeError(
     68             f"Expected type of scale is Real|Variable, but got {type(scale)}"
     69         )

TypeError: Expected type of loc is Real|Variable, but got <class 'complex'>

[ooooo-create]

是的，已修改~

[megemini]

Suggested change

	loc: Numberic
	scale: Numberic
	loc: Tensor
	scale: Tensor

实例化之后，loc scale 的值都转换为了 Tensor

[SigureMo]

LGTMeow