Skip to content

Commit

Permalink
removed redundant code from test_inference
Browse files Browse the repository at this point in the history
  • Loading branch information
rfl-urbaniak committed Mar 13, 2024
1 parent a0c0526 commit 9fe9130
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions tests/test_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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")
Expand Down

0 comments on commit 9fe9130

Please sign in to comment.