Skip to content

GitLab

Commit 67370881 authored by Andrea Paris's avatar Andrea Paris Committed by Boris Bonev
Browse files

correction

parent 9fb288c0
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
tests/test_sht.py
View file @ 67370881
...@@ -81,10 +81,9 @@ class TestLegendrePolynomials(unittest.TestCase): ...@@ -81,10 +81,9 @@ class TestLegendrePolynomials(unittest.TestCase):
class TestSphericalHarmonicTransform(unittest.TestCase): class TestSphericalHarmonicTransform(unittest.TestCase):
"""Test the spherical harmonic transform (CPU/CUDA if available).""" """Test the spherical harmonic transform (CPU/CUDA if available)."""
def setUp(self): def setUp(self):
if torch.cuda.is_available(): torch.manual_seed(333)
self.device = torch.device("cuda") if self.device.type == "cuda":
else: torch.cuda.manual_seed(333)
self.device = torch.device("cpu")
@parameterized.expand( @parameterized.expand(
[ [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment