Make spherical harmonics test more forgiving (#2607)

This commit is contained in:
Brent Yi authored and GitHub committed 2023-11-14 00:00:21 -08:00
1 parent 5f5562b191
commit 1f5f0c792b
1 file changed
+1 -1
+1 -1
View File
@@ -13,4 +13,4 @@ def test_spherical_harmonics(components):
dx = dx / torch.linalg.norm(dx, dim=-1, keepdim=True)
sh = components_from_spherical_harmonics(components, dx)
matrix = (sh.T @ sh) / N * 4 * torch.pi
torch.testing.assert_close(matrix, torch.eye(components**2), rtol=0, atol=1e-2)
torch.testing.assert_close(matrix, torch.eye(components**2), rtol=0, atol=1.5e-2)