Skip to content

Commit

Permalink
black reformatting
Browse files Browse the repository at this point in the history
  • Loading branch information
allaffa committed Apr 14, 2023
1 parent 7ccd7bf commit 2bcd1e1
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tests/test_graphs.py
Original file line number Diff line number Diff line change
Expand Up @@ -189,6 +189,8 @@ def pytest_train_model(model_type, skip_connection, ci_input, overwrite_data=Fal
@pytest.mark.parametrize("model_type", ["PNA", "CGCNN", "SchNet"])
def pytest_train_model_lengths(model_type, overwrite_data=False):
unittest_train_model(model_type, "ci.json", True, overwrite_data)


@pytest.mark.parametrize("skip_connection", [False, True])
def pytest_train_model_lengths(model_type, skip_connection, overwrite_data=False):
unittest_train_model(model_type, skip_connection, "ci.json", True, overwrite_data)
Expand Down

0 comments on commit 2bcd1e1

Please sign in to comment.