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 Dec 19, 2024
1 parent ff5bbf0 commit b7923f1
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 0 deletions.
1 change: 1 addition & 0 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -157,6 +157,7 @@ def sample_pv_batch():
# old batches. For now we use the old batches to test the site encoder models
return torch.load("tests/test_data/presaved_batches/train/000000.pt")


# TODO update this test once we add the loading logic for the Site dataset
# @pytest.fixture()
# def sample_wind_batch():
Expand Down
1 change: 1 addition & 0 deletions tests/models/multimodal/site_encoders/test_encoders.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@ def test_singleattentionnetwork_forward(sample_pv_batch, site_encoder_model_kwar
batch_size=8,
)


# TODO once we have updated the sample batches for sites include this test
# def test_singleattentionnetwork_forward_4d(sample_wind_batch, site_encoder_sensor_model_kwargs):
# _test_model_forward(
Expand Down

0 comments on commit b7923f1

Please sign in to comment.