提交 5d48ac0c authored 作者: Frederic's avatar Frederic

Fix tests since change in tests interface

上级 fb76c64b
...@@ -377,7 +377,7 @@ class G_Join_and_Split(test_basic.T_Join_and_Split): ...@@ -377,7 +377,7 @@ class G_Join_and_Split(test_basic.T_Join_and_Split):
m = self.shared(rng.rand(4, 6).astype(self.floatX)) m = self.shared(rng.rand(4, 6).astype(self.floatX))
o = T.Split(2)(m, 0, [2, 2]) o = T.Split(2)(m, 0, [2, 2])
f = theano.function([], o, mode=self.mode) f = theano.function([], o, mode=self.mode)
assert any([isinstance(node.op, self.split_op) assert any([isinstance(node.op, self.split_op_class)
for node in f.maker.fgraph.toposort()]) for node in f.maker.fgraph.toposort()])
o1, o2 = f() o1, o2 = f()
assert numpy.allclose(o1, m.get_value(borrow=True)[:2]) assert numpy.allclose(o1, m.get_value(borrow=True)[:2])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论