Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Feb 21, 2025
1 parent d35cb58 commit bc3a9f9
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion tests/integration/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,17 @@
from omegaconf import OmegaConf


@pytest.fixture(params=[["model=gnn"], ["model=graphtransformer"], ["model=transformer", "graph=encoder_decoder_only", "model.processor.attention_implementation=scaled_dot_product_attention"]])
@pytest.fixture(
params=[
["model=gnn"],
["model=graphtransformer"],
[
"model=transformer",
"graph=encoder_decoder_only",
"model.processor.attention_implementation=scaled_dot_product_attention",
],
]
)
def architecture_config(request) -> None:
overrides = request.param
with initialize(version_base=None, config_path="", job_name="test_training"):
Expand Down

0 comments on commit bc3a9f9

Please sign in to comment.