PVNet 5.1.1__tar.gz → 5.1.2__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 (36) hide show
  1. {pvnet-5.1.1 → pvnet-5.1.2}/PKG-INFO +1 -1
  2. {pvnet-5.1.1 → pvnet-5.1.2}/PVNet.egg-info/PKG-INFO +1 -1
  3. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/utils.py +13 -1
  4. {pvnet-5.1.1 → pvnet-5.1.2}/LICENSE +0 -0
  5. {pvnet-5.1.1 → pvnet-5.1.2}/PVNet.egg-info/SOURCES.txt +0 -0
  6. {pvnet-5.1.1 → pvnet-5.1.2}/PVNet.egg-info/dependency_links.txt +0 -0
  7. {pvnet-5.1.1 → pvnet-5.1.2}/PVNet.egg-info/requires.txt +0 -0
  8. {pvnet-5.1.1 → pvnet-5.1.2}/PVNet.egg-info/top_level.txt +0 -0
  9. {pvnet-5.1.1 → pvnet-5.1.2}/README.md +0 -0
  10. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/__init__.py +0 -0
  11. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/datamodule.py +0 -0
  12. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/load_model.py +0 -0
  13. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/__init__.py +0 -0
  14. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/base_model.py +0 -0
  15. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/ensemble.py +0 -0
  16. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/__init__.py +0 -0
  17. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/basic_blocks.py +0 -0
  18. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/encoders/__init__.py +0 -0
  19. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/encoders/basic_blocks.py +0 -0
  20. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/encoders/encoders3d.py +0 -0
  21. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/late_fusion.py +0 -0
  22. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/linear_networks/__init__.py +0 -0
  23. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/linear_networks/basic_blocks.py +0 -0
  24. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/linear_networks/networks.py +0 -0
  25. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/site_encoders/__init__.py +0 -0
  26. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/site_encoders/basic_blocks.py +0 -0
  27. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/models/late_fusion/site_encoders/encoders.py +0 -0
  28. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/optimizers.py +0 -0
  29. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/training/__init__.py +0 -0
  30. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/training/lightning_module.py +0 -0
  31. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/training/plots.py +0 -0
  32. {pvnet-5.1.1 → pvnet-5.1.2}/pvnet/training/train.py +0 -0
  33. {pvnet-5.1.1 → pvnet-5.1.2}/pyproject.toml +0 -0
  34. {pvnet-5.1.1 → pvnet-5.1.2}/setup.cfg +0 -0
  35. {pvnet-5.1.1 → pvnet-5.1.2}/tests/test_datamodule.py +0 -0
  36. {pvnet-5.1.1 → pvnet-5.1.2}/tests/test_end2end.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: PVNet
3
- Version: 5.1.1
3
+ Version: 5.1.2
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.1.1
3
+ Version: 5.1.2
4
4
  Summary: PVNet
5
5
  Author-email: Peter Dudfield <info@openclimatefix.org>
6
6
  Requires-Python: >=3.11
@@ -21,7 +21,6 @@ FULL_CONFIG_NAME = "full_experiment_config.yaml"
21
21
  MODEL_CARD_NAME = "README.md"
22
22
 
23
23
 
24
-
25
24
  def run_config_utilities(config: DictConfig) -> None:
26
25
  """A couple of optional utilities.
27
26
 
@@ -155,3 +154,16 @@ def validate_batch_against_config(
155
154
  )
156
155
 
157
156
  logger.info("Batch shape validation successful!")
157
+
158
+
159
+ def validate_gpu_config(config: DictConfig) -> None:
160
+ """Abort if multiple GPUs requested by mistake i.e. `devices: 2` instead of `[2]`."""
161
+ tr = config.get("trainer", {})
162
+ dev = tr.get("devices")
163
+
164
+ if isinstance(dev, int) and dev > 1:
165
+ raise ValueError(
166
+ f"Detected `devices: {dev}` — this requests {dev} GPUs. "
167
+ "If you meant a specific GPU (e.g. GPU 2), use `devices: [2]`. "
168
+ "Parallel training not supported."
169
+ )
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
File without changes
File without changes
File without changes
File without changes