提交 24c6dc56 authored 作者: abalkin's avatar abalkin

Issue #1193: Fixed theano-nose --batch.

上级 2638c332
...@@ -87,6 +87,7 @@ def main(): ...@@ -87,6 +87,7 @@ def main():
if time_prof_args or batch_args: if time_prof_args or batch_args:
from theano.tests import run_tests_in_batch from theano.tests import run_tests_in_batch
return run_tests_in_batch.main( return run_tests_in_batch.main(
theano_nose=os.path.realpath(__file__),
batch_size=batch_size, batch_size=batch_size,
time_profile=bool(time_prof_args), time_profile=bool(time_prof_args),
display_batch_output=display_batch_output) display_batch_output=display_batch_output)
......
...@@ -101,7 +101,7 @@ def main(stdout=None, stderr=None, argv=None, theano_nose=None, ...@@ -101,7 +101,7 @@ def main(stdout=None, stderr=None, argv=None, theano_nose=None,
theano_nose = path theano_nose = path
break break
if theano_nose is None: if theano_nose is None:
raise Exception("Not able to find theano_nose") raise Exception("Not able to find theano-nose")
if batch_size is None: if batch_size is None:
batch_size = 100 batch_size = 100
stdout_backup = sys.stdout stdout_backup = sys.stdout
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论