PVNet 5.0.17__tar.gz → 5.0.18__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (38) hide show
  1. {pvnet-5.0.17 → pvnet-5.0.18}/PKG-INFO +1 -1
  2. {pvnet-5.0.17 → pvnet-5.0.18}/PVNet.egg-info/PKG-INFO +1 -1
  3. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/encoders/basic_blocks.py +1 -0
  4. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/encoders/encoders3d.py +3 -3
  5. {pvnet-5.0.17 → pvnet-5.0.18}/LICENSE +0 -0
  6. {pvnet-5.0.17 → pvnet-5.0.18}/PVNet.egg-info/SOURCES.txt +0 -0
  7. {pvnet-5.0.17 → pvnet-5.0.18}/PVNet.egg-info/dependency_links.txt +0 -0
  8. {pvnet-5.0.17 → pvnet-5.0.18}/PVNet.egg-info/requires.txt +0 -0
  9. {pvnet-5.0.17 → pvnet-5.0.18}/PVNet.egg-info/top_level.txt +0 -0
  10. {pvnet-5.0.17 → pvnet-5.0.18}/README.md +0 -0
  11. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/__init__.py +0 -0
  12. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/data/__init__.py +0 -0
  13. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/data/base_datamodule.py +0 -0
  14. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/data/site_datamodule.py +0 -0
  15. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/data/uk_regional_datamodule.py +0 -0
  16. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/load_model.py +0 -0
  17. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/__init__.py +0 -0
  18. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/base_model.py +0 -0
  19. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/ensemble.py +0 -0
  20. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/__init__.py +0 -0
  21. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/basic_blocks.py +0 -0
  22. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/encoders/__init__.py +0 -0
  23. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/late_fusion.py +0 -0
  24. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/linear_networks/__init__.py +0 -0
  25. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/linear_networks/basic_blocks.py +0 -0
  26. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/linear_networks/networks.py +0 -0
  27. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/site_encoders/__init__.py +0 -0
  28. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/site_encoders/basic_blocks.py +0 -0
  29. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/models/late_fusion/site_encoders/encoders.py +0 -0
  30. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/optimizers.py +0 -0
  31. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/training/__init__.py +0 -0
  32. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/training/lightning_module.py +0 -0
  33. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/training/plots.py +0 -0
  34. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/training/train.py +0 -0
  35. {pvnet-5.0.17 → pvnet-5.0.18}/pvnet/utils.py +0 -0
  36. {pvnet-5.0.17 → pvnet-5.0.18}/pyproject.toml +0 -0
  37. {pvnet-5.0.17 → pvnet-5.0.18}/setup.cfg +0 -0
  38. {pvnet-5.0.17 → pvnet-5.0.18}/tests/test_end2end.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: PVNet
3
- Version: 5.0.17
3
+ Version: 5.0.18
4
4
  Summary: PVNet
5
5
  Author-email: Peter Dudfield <info@openclimatefix.org>
6
6
  Requires-Python: >=3.11
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: PVNet
3
- Version: 5.0.17
3
+ Version: 5.0.18
4
4
  Summary: PVNet
5
5
  Author-email: Peter Dudfield <info@openclimatefix.org>
6
6
  Requires-Python: >=3.11
@@ -16,6 +16,7 @@ class AbstractNWPSatelliteEncoder(nn.Module, metaclass=ABCMeta):
16
16
  self,
17
17
  sequence_length: int,
18
18
  image_size_pixels: int,
19
+ in_channels: int,
19
20
  out_features: int,
20
21
  ):
21
22
  """Abstract class for NWP/satellite encoder.
@@ -24,7 +24,7 @@ class DefaultPVNet(AbstractNWPSatelliteEncoder):
24
24
  fc_features: int = 128,
25
25
  spatial_kernel_size: int = 3,
26
26
  temporal_kernel_size: int = 3,
27
- padding: int | tuple[int] = (1, 0, 0),
27
+ padding: int | tuple[int, ...] = (1, 0, 0),
28
28
  ):
29
29
  """This is the original encoding module used in PVNet, with a few minor tweaks.
30
30
 
@@ -42,7 +42,7 @@ class DefaultPVNet(AbstractNWPSatelliteEncoder):
42
42
  is used in all dimensions
43
43
  """
44
44
 
45
- super().__init__(sequence_length, image_size_pixels, out_features)
45
+ super().__init__(sequence_length, image_size_pixels, in_channels, out_features)
46
46
 
47
47
  if isinstance(padding, int):
48
48
  padding = (padding, padding, padding)
@@ -137,7 +137,7 @@ class ResConv3DNet(AbstractNWPSatelliteEncoder):
137
137
  batch_norm: Whether to include batch normalisation.
138
138
  dropout_frac: Probability of an element to be zeroed in the residual pathways.
139
139
  """
140
- super().__init__(sequence_length, image_size_pixels, out_features)
140
+ super().__init__(sequence_length, image_size_pixels, in_channels, out_features)
141
141
 
142
142
  model = [
143
143
  nn.Conv3d(
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes