correction

67370881 · Andrea Paris · Boris Bonev · 9fb288c0 · 67370881
Commit 67370881 authored Jul 21, 2025 by Andrea Paris Committed by Boris Bonev Jul 21, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

tests/test_sht.py tests/test_sht.py +3 -4

No files found.
--- a/tests/test_sht.py
+++ b/tests/test_sht.py
@@ -81,10 +81,9 @@ class TestLegendrePolynomials(unittest.TestCase):
 class TestSphericalHarmonicTransform(unittest.TestCase):
    """Test the spherical harmonic transform (CPU/CUDA if available)."""
    def setUp(self):
-        if torch.cuda.is_available():
-            self.device = torch.device("cuda")
-        else:
-            self.device = torch.device("cpu")
+        torch.manual_seed(333)
+        if self.device.type == "cuda":
+            torch.cuda.manual_seed(333)

    @parameterized.expand(
        [