提交 15f30cda authored 作者: --global's avatar --global

PEP8

上级 536d3b00
......@@ -2,10 +2,10 @@ import numpy
import pdb
import theano
import theano.tensor as T
from theano.gof import Op, Apply
from theano.gradient import DisconnectedType
class PdbBreakpoint(Op):
"""
This is an identity-like op with the side effect of enforcing a
......
......@@ -4,6 +4,7 @@ import theano.tensor as T
from theano.tests import unittest_tools as utt
from theano.tests.breakpoint import PdbBreakpoint
class TestPdbBreakpoint(utt.InferShapeTester):
def setUp(self):
......@@ -28,7 +29,7 @@ class TestPdbBreakpoint(utt.InferShapeTester):
def test_infer_shape(self):
input1_value = numpy.arange(6).reshape(2,3).astype("float32")
input1_value = numpy.arange(6).reshape(2, 3).astype("float32")
input2_value = 10.0
self._compile_and_check([self.input1, self.input2],
......@@ -40,7 +41,7 @@ class TestPdbBreakpoint(utt.InferShapeTester):
def test_grad(self):
input1_value = numpy.arange(9).reshape(3,3).astype("float32")
input1_value = numpy.arange(9).reshape(3, 3).astype("float32")
input2_value = 10.0
grads = [T.grad(self.monitored_input1.sum(), self.input1),
......@@ -62,7 +63,7 @@ class TestPdbBreakpoint(utt.InferShapeTester):
def test_fprop(self):
input1_value = numpy.arange(9).reshape(3,3).astype("float32")
input1_value = numpy.arange(9).reshape(3, 3).astype("float32")
input2_value = 10.0
fct = theano.function([self.input1, self.input2],
[self.monitored_input1, self.monitored_input2])
......@@ -75,6 +76,6 @@ class TestPdbBreakpoint(utt.InferShapeTester):
node = self.monitored_output.owner
connection_pattern = self.breakpointOp.connection_pattern(node)
expected_pattern = [[0,0,0],[1,0,0],[0,1,0],[0,0,1]]
expected_pattern = [[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]]
assert connection_pattern == expected_pattern
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论