Commit 4db0398a authored by vfdev's avatar vfdev Committed by Soumith Chintala
Browse files

Update test_utils.py (#486)

Remove test with variables
parent f87a896f
......@@ -18,14 +18,6 @@ class Tester(unittest.TestCase):
utils.make_grid(t, normalize=True, scale_each=True)
assert torch.equal(t, t_clone), 'make_grid modified tensor in-place'
def test_make_grid_raises_with_variable(self):
t = torch.autograd.Variable(torch.rand(3, 10, 10))
with self.assertRaises(TypeError):
utils.make_grid(t)
with self.assertRaises(TypeError):
utils.make_grid([t, t, t, t])
if __name__ == '__main__':
unittest.main()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment