提交 12ef382b authored 作者: zoj's avatar zoj 提交者: Thomas Wiecki

Use a Class-base solution instead of functools.partial to expose standard_normal

上级 2b61312b
import abc import abc
import functools
from typing import List, Optional, Union from typing import List, Optional, Union
import numpy as np import numpy as np
...@@ -118,11 +117,14 @@ class NormalRV(RandomVariable): ...@@ -118,11 +117,14 @@ class NormalRV(RandomVariable):
normal = NormalRV() normal = NormalRV()
# used as an alias of normal(loc=0, scale=1) in order to be consistent with np.random.RandomState
standard_normal = functools.update_wrapper(
functools.partial(normal, loc=0.0, scale=1.0), class StandardNormalRV(NormalRV):
normal def __call__(self, size=None, **kwargs):
) return super().__call__(loc=0.0, scale=1.0, size=size, **kwargs)
standard_normal = StandardNormalRV()
class HalfNormalRV(ScipyRandomVariable): class HalfNormalRV(ScipyRandomVariable):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论