提交 05378c82 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Print warning only once

上级 9120fd0d
...@@ -13,6 +13,7 @@ import operator ...@@ -13,6 +13,7 @@ import operator
import sys import sys
import time import time
import traceback import traceback
import warnings
import numpy import numpy
import numpy as N # guys... please don't do this in the library :( import numpy as N # guys... please don't do this in the library :(
...@@ -3908,7 +3909,7 @@ def local_reduce_join(node): ...@@ -3908,7 +3909,7 @@ def local_reduce_join(node):
# I put this warning late to don't add extra warning. # I put this warning late to don't add extra warning.
if len(reduce_axis) != 1 or 0 not in reduce_axis: if len(reduce_axis) != 1 or 0 not in reduce_axis:
if theano.config.warn.reduce_join: if theano.config.warn.reduce_join:
_logger.warn(( warnings.warn((
'Your current code is fine, but Theano versions ' 'Your current code is fine, but Theano versions '
'prior to 0.7 (or this development version Sept 2014) ' 'prior to 0.7 (or this development version Sept 2014) '
'might have given an incorrect result for this code. ' 'might have given an incorrect result for this code. '
......
...@@ -4447,7 +4447,8 @@ class T_local_reduce(unittest.TestCase): ...@@ -4447,7 +4447,8 @@ class T_local_reduce(unittest.TestCase):
assert isinstance(topo[-1].op, T.Elemwise), out assert isinstance(topo[-1].op, T.Elemwise), out
# Test different axis for the join and the reduction # Test different axis for the join and the reduction
# We must force the dtype, of otherwise, this tests will fail in 32 bit system # We must force the dtype, of otherwise, this tests will fail
# on 32 bit systems
A = theano.shared(numpy.array([1, 2, 3, 4, 5], dtype='int64')) A = theano.shared(numpy.array([1, 2, 3, 4, 5], dtype='int64'))
f = theano.function([], T.sum(T.stack(A, A), axis=0), mode=self.mode) f = theano.function([], T.sum(T.stack(A, A), axis=0), mode=self.mode)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论