From 1fab3f4f7086e694a07d98a37f5423eb6ec05889 Mon Sep 17 00:00:00 2001
From: Agustinus Kristiadi <agustinus@kristia.de>
Date: Sun, 15 Sep 2024 10:29:13 -0400
Subject: [PATCH] Specify `torch.double` as the default in
 `test_functional_laplace*.py`

---
 tests/test_functional_laplace.py      | 2 ++
 tests/test_functional_laplace_unit.py | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/tests/test_functional_laplace.py b/tests/test_functional_laplace.py
index 82c0247..f0a4e03 100644
--- 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
index 783af0e..4120b59 100644
--- 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():