提交 cdd8bf2c authored 作者: Saizheng Zhang's avatar Saizheng Zhang

fix flake8 error in test_broadcastable

上级 2455dc36
......@@ -633,13 +633,13 @@ def test_broadcastable():
size1 = (10, 1)
size2 = (x.shape[0], 1)
pvals_1 = numpy.random.uniform(0, 1, size=size1)
pvals_1 = pvals_1/sum(pvals_1)
pvals_1 = pvals_1 / sum(pvals_1)
pvals_2 = R.uniform(size=size2)
pvals_2 = pvals_2/tensor.sum(pvals_2)
pvals_2 = pvals_2 / tensor.sum(pvals_2)
for distribution in [R.uniform, R.binomial, R.multinomial, R.multinomial_wo_replacement, R.normal]:
# multinomial or multinomial_wo_replacement does not support "size" argument,
# the size of them are implicitly defined with "pvals" argument.
# the sizes of them are implicitly defined with "pvals" argument.
if distribution in [R.multinomial, R.multinomial_wo_replacement]:
# check when all dimensions are constant
uu = distribution(pvals=pvals_1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论