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")