Update test_rand_deform_grid.py

Author: wyli

tests/test_rand_deform_grid.py

@@ -129,9 +129,8 @@ def test_rand_deform_grid(self, input_param, input_data, expected_val):
         g = RandDeformGrid(**input_param)
         g.set_random_state(123)
         result = g(**input_data)
-        self.assertEqual(isinstance(result, torch.Tensor), isinstance(expected_val, torch.Tensor))
         if isinstance(result, torch.Tensor):
-            np.testing.assert_allclose(result.cpu().numpy(), expected_val.cpu().numpy(), rtol=1e-4, atol=1e-4)
+            np.testing.assert_allclose(result.cpu().numpy(), expected_val, rtol=1e-4, atol=1e-4)
         else:
             np.testing.assert_allclose(result, expected_val, rtol=1e-4, atol=1e-4)