From 28ff98291a27101cc6b42437707468863048478b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Paul-Christian=20B=C3=BCrkner?= Date: Thu, 10 Oct 2024 07:56:59 +0200 Subject: [PATCH] fix tests for student_t --- tests/test_distributions/conftest.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_distributions/conftest.py b/tests/test_distributions/conftest.py index 8734b889d..20e4b3c4f 100644 --- 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,6 +20,8 @@ def random_samples(batch_size, num_features): @pytest.fixture() def diagonal_normal(): + from bayesflow.distributions import DiagonalNormal + return DiagonalNormal() @@ -29,7 +29,7 @@ def diagonal_normal(): def diagonal_student_t(): from bayesflow.distributions import DiagonalStudentT - return DiagonalStudentT() + return DiagonalStudentT(df = 10) @pytest.fixture(params=["diagonal_normal", "diagonal_student_t"])