More flake8 fixes

上级 e8ef7b16
...@@ -282,8 +282,7 @@ class GpuCublasTriangularSolve(Op): ...@@ -282,8 +282,7 @@ class GpuCublasTriangularSolve(Op):
assert inp1.dtype == 'float32' assert inp1.dtype == 'float32'
assert inp2.dtype == 'float32' assert inp2.dtype == 'float32'
return theano.Apply( return theano.Apply(self, [inp1, inp2],
self, [inp1, inp2],
[GpuArrayType('float32', [GpuArrayType('float32',
broadcastable=inp2.broadcastable, broadcastable=inp2.broadcastable,
context_name=context_name)()]) context_name=context_name)()])
...@@ -305,7 +304,7 @@ class GpuCublasTriangularSolve(Op): ...@@ -305,7 +304,7 @@ class GpuCublasTriangularSolve(Op):
b = inputs[1] b = inputs[1]
assert(len(A.shape) == 2) assert(len(A.shape) == 2)
assert(len(b.shape) in [1,2]) assert(len(b.shape) in [1, 2])
# implicitly deal with the difference between C order # implicitly deal with the difference between C order
# and fortran order by flipping the trans and lower flags # and fortran order by flipping the trans and lower flags
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论