提交 75150064 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Got rid of another duplicated list of dtypes

上级 5b498c64
...@@ -13,7 +13,6 @@ from theano.compile.mode import get_default_mode ...@@ -13,7 +13,6 @@ from theano.compile.mode import get_default_mode
from theano.tensor.elemwise import * from theano.tensor.elemwise import *
from theano.tests import unittest_tools from theano.tests import unittest_tools
complex_dtypes = map(str, scalar.complex_types)
def Env(i, o): def Env(i, o):
e = gof.Env(i, o) e = gof.Env(i, o)
...@@ -594,7 +593,7 @@ class T_mean_dtype(unittest.TestCase): ...@@ -594,7 +593,7 @@ class T_mean_dtype(unittest.TestCase):
except NotImplementedError: except NotImplementedError:
# TrueDiv does not seem to have a gradient when # TrueDiv does not seem to have a gradient when
# the numerator is complex. # the numerator is complex.
if mean_var.dtype in complex_dtypes: if mean_var.dtype in tensor.complex_dtypes:
pass pass
else: else:
raise raise
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论