提交 952fd60a authored 作者: Frederic Bastien's avatar Frederic Bastien

accept DebugMode as synonim of DEBUG_MODE

上级 b2d0f53d
......@@ -321,7 +321,7 @@ def test_uniform():
#TODO: test ndim!=size.ndim
#TODO: test bad seed
#TODO: test size=Var, with shape that change from call to call
if mode in ['DEBUG_MODE','FAST_COMPILE']:
if mode in ['DEBUG_MODE','DebugMode','FAST_COMPILE']:
sample_size = (10,100)
steps = 50
else:
......@@ -380,7 +380,7 @@ def test_binomial():
#we test size in a tuple of int and a tensor.shape.
#we test the param p with int.
if mode in ['DEBUG_MODE','FAST_COMPILE']:
if mode in ['DEBUG_MODE','DebugMode','FAST_COMPILE']:
sample_size = (10,50)
steps = 50
rtol=0.02
......@@ -432,7 +432,7 @@ def test_binomial():
def test_normal0():
steps = 50
if mode in ['DEBUG_MODE','FAST_COMPILE']:
if mode in ['DEBUG_MODE','DebugMode','FAST_COMPILE']:
sample_size = (25,30)
rtol=.02
else:
......@@ -511,7 +511,7 @@ def test_multinomial():
if mode == 'FAST_COMPILE':
mode_ = 'FAST_RUN'
if mode in ['DEBUG_MODE','FAST_COMPILE']:
if mode in ['DEBUG_MODE','DebugMode','FAST_COMPILE']:
sample_size = (49,5)
else:
sample_size = (450,6)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论