diff --git a/python/featomic/tests/clebsch_gordan/equivariant_power_spectrum.py b/python/featomic/tests/clebsch_gordan/equivariant_power_spectrum.py index 9cd504d89..1f47efe64 100644 --- a/python/featomic/tests/clebsch_gordan/equivariant_power_spectrum.py +++ b/python/featomic/tests/clebsch_gordan/equivariant_power_spectrum.py @@ -15,7 +15,6 @@ import ase.io # noqa: E402, F811 -MAX_ANGULAR = 2 SPHEX_HYPERS_SMALL = { "cutoff": { "radius": 2.5, @@ -27,7 +26,7 @@ }, "basis": { "type": "TensorProduct", - "max_angular": MAX_ANGULAR, + "max_angular": 2, "radial": {"type": "Gto", "max_radial": 1}, }, } diff --git a/python/featomic_torch/tests/clebsch_gordan/equivariant_power_spectrum.py b/python/featomic_torch/tests/clebsch_gordan/equivariant_power_spectrum.py index ab2fddcf8..d4af57a30 100644 --- a/python/featomic_torch/tests/clebsch_gordan/equivariant_power_spectrum.py +++ b/python/featomic_torch/tests/clebsch_gordan/equivariant_power_spectrum.py @@ -6,10 +6,9 @@ from featomic.torch.clebsch_gordan import EquivariantPowerSpectrum -MAX_ANGULAR = 2 SPHEX_HYPERS_SMALL = { "cutoff": { - "radius": 2.5, + "radius": 5.5, "smoothing": {"type": "ShiftedCosine", "width": 0.5}, }, "density": { @@ -18,10 +17,8 @@ }, "basis": { "type": "TensorProduct", - # use a small basis to make the tests faster - # FIXME: setting max_angular=1 breaks the tests - "max_angular": MAX_ANGULAR, - "radial": {"type": "Gto", "max_radial": 1}, + "max_angular": 6, + "radial": {"type": "Gto", "max_radial": 4}, }, }