From feb5d36394b28e382e9a58e670fb5a92684bb4bb Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Mon, 21 Feb 2022 06:32:09 -0800 Subject: [PATCH] points2vols test fix Summary: Fix tests which depended on output tensors being identical to input ones, which now fail in main PyTorch branch because of some change in autograd. The functions still work in-place. Reviewed By: patricklabatut Differential Revision: D34375817 fbshipit-source-id: 295ae195f75eab6c7abab412c997470d8de8add1 --- tests/test_points_to_volumes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_points_to_volumes.py b/tests/test_points_to_volumes.py index 11563729..7284b587 100644 --- a/tests/test_points_to_volumes.py +++ b/tests/test_points_to_volumes.py @@ -567,8 +567,8 @@ class TestRawFunction(TestCaseMixin, unittest.TestCase): splat, ) - self.assertIs(volume_densities, volume_densities_) - self.assertIs(volume_features, volume_features_) + self.assertTrue(volume_densities_.is_set_to(volume_densities)) + self.assertTrue(volume_features_.is_set_to(volume_features)) if align_corners: volume_densities_expected[0, 0, 2, 3, 4] = point_weight