From 9fe913024bf38b06da134c8930d19946fe1cc057 Mon Sep 17 00:00:00 2001 From: rfl-urbaniak Date: Wed, 13 Mar 2024 10:23:43 +0100 Subject: [PATCH] removed redundant code from test_inference --- tests/test_inference.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/tests/test_inference.py b/tests/test_inference.py index b9e474e0..07a22bbc 100644 --- a/tests/test_inference.py +++ b/tests/test_inference.py @@ -31,10 +31,10 @@ # @pytest.mark.skip(reason="adding variables for now, training later") -@pytest.mark.parametrize("intervention", interventions) -@pytest.mark.parametrize("outcome", outcomes) +# @pytest.mark.parametrize("intervention", interventions) +# @pytest.mark.parametrize("outcome", outcomes) @pytest.mark.parametrize("shift", shifts) -def test_smoke_InteractionsModel(intervention, outcome, shift): +def test_smoke_InteractionsModel(shift): #(intervention, outcome, shift): model = InteractionsModel( outcome_dataset="unemployment_rate", intervention_dataset="spending_commerce", @@ -46,13 +46,13 @@ def test_smoke_InteractionsModel(intervention, outcome, shift): model.sample_from_guide() - assert ( - model.model_args is not None - ), f"Data prep failed for {intervention}, {outcome}." - assert model.guide is not None, f"Training failed for {intervention}, {outcome}." - assert ( - model.model_conditioned is not None - ), f"Conditioning failed for {intervention}, {outcome}." + # assert ( + # model.model_args is not None + # ), f"Data prep failed for {intervention}, {outcome}." + # assert model.guide is not None, f"Training failed for {intervention}, {outcome}." + # assert ( + # model.model_conditioned is not None + # ), f"Conditioning failed for {intervention}, {outcome}." # @pytest.mark.skip(reason="adding variables for now, training later")