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 13, 2025
1 parent e057400 commit 0f52653
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 4 deletions.
6 changes: 3 additions & 3 deletions pvnet/data/base_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,12 @@

from glob import glob
from typing import Type

from lightning.pytorch import LightningDataModule
from ocf_data_sampler.numpy_sample.collate import stack_np_samples_into_batch

from ocf_data_sampler.sample.base import (
SampleBase,
NumpyBatch,
SampleBase,
TensorBatch,
batch_to_tensor,
)
Expand All @@ -28,7 +28,7 @@ class PremadeSamplesDataset(Dataset):
"""

def __init__(self, sample_dir: str, sample_class: Type[SampleBase]):
"""Initialise PremadeSamplesDataset"""
"""Initialise PremadeSamplesDataset"""
self.sample_paths = glob(f"{sample_dir}/*")
self.sample_class = sample_class

Expand Down
1 change: 0 additions & 1 deletion tests/data/test_datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,4 +72,3 @@ def test_site_init_config():
val_period=[None, None],
sample_dir=None,
)

0 comments on commit 0f52653

Please sign in to comment.