Skip to content

Commit

Permalink
resolved #302
Browse files Browse the repository at this point in the history
  • Loading branch information
hvgazula committed Mar 21, 2024
1 parent 4d67696 commit da5e0bc
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions nobrainer/models/tests/models_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -266,6 +266,10 @@ def test_unetr():
model_test(unetr, n_classes=1, input_shape=(1, 96, 96, 96, 1))


@pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Cannot test in GitHub Actions")
def test_variational_meshnet():
model_test(variational_meshnet, n_classes=1, input_shape=(1, 128, 128, 128, 1))
model_test(
variational_meshnet,
n_classes=1,
input_shape=(1, 128, 128, 128, 1),
kwds={"filters": 4},
)

0 comments on commit da5e0bc

Please sign in to comment.