Specify torch.double as the default in test_functional_laplace*.py

aleximmer · Sep 15, 2024 · 1fab3f4 · 1fab3f4
1 parent 2571a7e
commit 1fab3f4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/tests/test_functional_laplace.py b/tests/test_functional_laplace.py
@@ -20,7 +20,9 @@
 )
 
 true_sigma_noise = 0.1
+
 torch.manual_seed(711)
+torch.set_default_dtype(torch.double)
 
 
 @pytest.mark.parametrize(

diff --git a/tests/test_functional_laplace_unit.py b/tests/test_functional_laplace_unit.py
@@ -6,6 +6,9 @@
 
 from laplace.baselaplace import FunctionalLaplace
 
+torch.manual_seed(240)
+torch.set_default_dtype(torch.double)
+
 
 @pytest.fixture
 def reg_loader():