fix tests for student_t

bayesflow-org · Oct 10, 2024 · 28ff982 · 28ff982
1 parent b73460c
commit 28ff982
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_distributions/conftest.py b/tests/test_distributions/conftest.py
@@ -2,8 +2,6 @@
 
 import keras
 
-from bayesflow.distributions import DiagonalNormal
-
 
 @pytest.fixture(params=[2, 3])
 def batch_size(request):
@@ -22,14 +20,16 @@ def random_samples(batch_size, num_features):
 
 @pytest.fixture()
 def diagonal_normal():
+    from bayesflow.distributions import DiagonalNormal
+
     return DiagonalNormal()
 
 
 @pytest.fixture()
 def diagonal_student_t():
     from bayesflow.distributions import DiagonalStudentT
 
-    return DiagonalStudentT()
+    return DiagonalStudentT(df = 10)
 
 
 @pytest.fixture(params=["diagonal_normal", "diagonal_student_t"])