Skip to content

Commit 2c287a8

Browse files
committed
Bugfix in test for quadratic data set
1 parent 08798e5 commit 2c287a8

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/datasets/test_quadratic.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,8 @@ def test_init_ops(self):
3232
self.quadratic.train_eval_init_op
3333
]:
3434
sess.run(init_op)
35-
x_, y_ = sess.run(self.quadratic.batch)
35+
x_ = sess.run(self.quadratic.batch)
3636
self.assertEqual(x_.shape, (self.batch_size, self.quadratic._dim))
37-
self.assertEqual(y_.shape, (self.batch_size, self.quadratic._dim))
3837

3938

4039
if __name__ == "__main__":

0 commit comments

Comments
 (0)