提交 99ed65a0 authored 作者: Iban Harlouchet's avatar Iban Harlouchet

flake8 for solve.py

上级 131ea96a
from __future__ import print_function from __future__ import print_function
import numpy, scipy.linalg
from theano import gof, tensor, scalar
import unittest import unittest
import sys
import numpy
import scipy.linalg
import theano
from theano import gof, tensor, scalar
from theano.tests import unittest_tools as utt
class Solve(gof.Op): class Solve(gof.Op):
...@@ -32,7 +39,7 @@ class Solve(gof.Op): ...@@ -32,7 +39,7 @@ class Solve(gof.Op):
raise TypeError("b must be a matrix or vector", b_.type) raise TypeError("b must be a matrix or vector", b_.type)
odtype = scalar.upcast(A_.dtype, b_.dtype) odtype = scalar.upcast(A_.dtype, b_.dtype)
otype = tensor.TensorType(broadcastable=b_.broadcastable, dtype=odtype) otype = tensor.TensorType(broadcastable=b_.broadcastable, dtype=odtype)
return gof.Apply(op=self, inputs=[A, B], outputs=[otype()]) return gof.Apply(op=self, inputs=[A_, b_], outputs=[otype()])
def perform(self, node, inp, out): def perform(self, node, inp, out):
A, b = inp A, b = inp
...@@ -49,8 +56,6 @@ solve = Solve() ...@@ -49,8 +56,6 @@ solve = Solve()
# TODO: test dtype conversion # TODO: test dtype conversion
# TODO: test that invalid types are rejected by make_node # TODO: test that invalid types are rejected by make_node
# TODO: test that each valid type for A and b works correctly # TODO: test that each valid type for A and b works correctly
from theano.tests import unittest_tools as utt
class T_solve(unittest.TestCase): class T_solve(unittest.TestCase):
def setUp(self): def setUp(self):
......
...@@ -135,7 +135,6 @@ whitelist_flake8 = [ ...@@ -135,7 +135,6 @@ whitelist_flake8 = [
"sandbox/test_theano_object.py", "sandbox/test_theano_object.py",
"sandbox/test_scan.py", "sandbox/test_scan.py",
"sandbox/rng_mrg.py", "sandbox/rng_mrg.py",
"sandbox/solve.py",
"sandbox/theano_object.py", "sandbox/theano_object.py",
"sandbox/scan.py", "sandbox/scan.py",
"sandbox/multinomial.py", "sandbox/multinomial.py",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论