aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/kernel_methods
diff options
context:
space:
mode:
authorGravatar Yifei Feng <yifeif@google.com>2018-01-26 16:53:59 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-26 16:59:01 -0800
commitaee7f95a027accc94f1f9130f0cfaecd9399bc1d (patch)
tree6b8484915bf631f18b2fa0561a73549d9bf19fad /tensorflow/contrib/kernel_methods
parente95537708f070a98607393a8f60bc61f1611a77b (diff)
Add C0301 line-too-long error to pylint sanity check.
PiperOrigin-RevId: 183467186
Diffstat (limited to 'tensorflow/contrib/kernel_methods')
-rw-r--r--tensorflow/contrib/kernel_methods/python/losses_test.py17
1 files changed, 9 insertions, 8 deletions
diff --git a/tensorflow/contrib/kernel_methods/python/losses_test.py b/tensorflow/contrib/kernel_methods/python/losses_test.py
index d38d8041ce..72507539f8 100644
--- a/tensorflow/contrib/kernel_methods/python/losses_test.py
+++ b/tensorflow/contrib/kernel_methods/python/losses_test.py
@@ -119,19 +119,20 @@ class SparseMulticlassHingeLossTest(test.TestCase):
def testUnknownShape(self):
"""Result keeps same with `testZeroLossInt32Labels`"""
- logits_np = np.array([[1.2, -1.4, -1.0],
- [1.4, 1.8, 4.0],
- [0.5, 1.8, -1.0]])
+ logits_np = np.array([[1.2, -1.4, -1.0], [1.4, 1.8, 4.0], [0.5, 1.8, -1.0]])
labels_np = np.array([0, 2, 1], dtype=np.int32)
- logits_shapes = [[3, 3], # batch_size, num_classes
- [None, 3],
- [3, None],
- [None, None]]
+ logits_shapes = [
+ [3, 3], # batch_size, num_classes
+ [None, 3],
+ [3, None],
+ [None, None]
+ ]
for batch_size, num_classes in logits_shapes:
with self.test_session():
- logits = array_ops.placeholder(dtypes.float32, shape=(batch_size, num_classes))
+ logits = array_ops.placeholder(
+ dtypes.float32, shape=(batch_size, num_classes))
labels = array_ops.placeholder(dtypes.int32, shape=(batch_size,))
loss = losses.sparse_multiclass_hinge_loss(labels, logits)
result = loss.eval(feed_dict={logits: logits_np, labels: labels_np})