ocf-data-sampler 0.6.0__tar.gz → 0.6.1__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.

Potentially problematic release.

This version of ocf-data-sampler might be problematic. Click here for more details.

Files changed (69) hide show
  1. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/PKG-INFO +1 -1
  2. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/site.py +14 -5
  3. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler.egg-info/PKG-INFO +1 -1
  4. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/LICENSE +0 -0
  5. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/README.md +0 -0
  6. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/__init__.py +0 -0
  7. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/config/__init__.py +0 -0
  8. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/config/load.py +0 -0
  9. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/config/model.py +0 -0
  10. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/config/save.py +0 -0
  11. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/data/uk_gsp_locations_20220314.csv +0 -0
  12. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/data/uk_gsp_locations_20250109.csv +0 -0
  13. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/__init__.py +0 -0
  14. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/gsp.py +0 -0
  15. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/load_dataset.py +0 -0
  16. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/__init__.py +0 -0
  17. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/nwp.py +0 -0
  18. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/providers/__init__.py +0 -0
  19. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/providers/cloudcasting.py +0 -0
  20. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/providers/ecmwf.py +0 -0
  21. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/providers/gfs.py +0 -0
  22. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/providers/icon.py +0 -0
  23. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/providers/ukv.py +0 -0
  24. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/nwp/providers/utils.py +0 -0
  25. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/open_xarray_tensorstore.py +0 -0
  26. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/satellite.py +0 -0
  27. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/utils.py +0 -0
  28. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/__init__.py +0 -0
  29. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/collate.py +0 -0
  30. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/common_types.py +0 -0
  31. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/datetime_features.py +0 -0
  32. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/gsp.py +0 -0
  33. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/nwp.py +0 -0
  34. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/satellite.py +0 -0
  35. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/site.py +0 -0
  36. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/numpy_sample/sun_position.py +0 -0
  37. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/__init__.py +0 -0
  38. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/diff_channels.py +0 -0
  39. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/dropout.py +0 -0
  40. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/fill_time_periods.py +0 -0
  41. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/find_contiguous_time_periods.py +0 -0
  42. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/geospatial.py +0 -0
  43. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/location.py +0 -0
  44. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/select_spatial_slice.py +0 -0
  45. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/select/select_time_slice.py +0 -0
  46. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/datasets/__init__.py +0 -0
  47. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/datasets/picklecache.py +0 -0
  48. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/datasets/pvnet_uk.py +0 -0
  49. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/datasets/site.py +0 -0
  50. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/__init__.py +0 -0
  51. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/add_alterate_coordinate_projections.py +0 -0
  52. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/config_normalization_values_to_dicts.py +0 -0
  53. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/diff_nwp_data.py +0 -0
  54. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/merge_and_fill_utils.py +0 -0
  55. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/spatial_slice_for_dataset.py +0 -0
  56. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/time_slice_for_dataset.py +0 -0
  57. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/torch_batch_utils.py +0 -0
  58. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/valid_time_periods.py +0 -0
  59. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/torch_datasets/utils/validation_utils.py +0 -0
  60. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/utils.py +0 -0
  61. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler.egg-info/SOURCES.txt +0 -0
  62. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler.egg-info/dependency_links.txt +0 -0
  63. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler.egg-info/requires.txt +0 -0
  64. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler.egg-info/top_level.txt +0 -0
  65. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/pyproject.toml +0 -0
  66. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/scripts/download_gsp_location_data.py +0 -0
  67. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/scripts/refactor_site.py +0 -0
  68. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/setup.cfg +0 -0
  69. {ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/tests/test_utils.py +0 -0
{ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ocf-data-sampler
3
- Version: 0.6.0
3
+ Version: 0.6.1
4
4
  Author: James Fulton, Peter Dudfield
5
5
  Author-email: Open Climate Fix team <info@openclimatefix.org>
6
6
  License: MIT License
{ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler/load/site.py RENAMED
@@ -25,11 +25,20 @@ def open_site(generation_file_path: str, metadata_file_path: str) -> xr.DataArra
25
25
  metadata_df = metadata_df.reindex(generation_ds.site_id.values)
26
26
 
27
27
  # Assign coordinates to the Dataset using the aligned metadata
28
- generation_ds = generation_ds.assign_coords(
29
- latitude=("site_id", metadata_df["latitude"].values),
30
- longitude=("site_id", metadata_df["longitude"].values),
31
- capacity_kwp=("site_id", metadata_df["capacity_kwp"].values),
32
- )
28
+ # Check if variable capacity was passed with the generation data
29
+ # If not assign static capacity from metadata
30
+ if hasattr(generation_ds,"capacity_kwp"):
31
+ generation_ds = generation_ds.assign_coords(
32
+ latitude=(metadata_df.latitude.to_xarray()),
33
+ longitude=(metadata_df.longitude.to_xarray()),
34
+ capacity_kwp=generation_ds.capacity_kwp,
35
+ )
36
+ else:
37
+ generation_ds = generation_ds.assign_coords(
38
+ latitude=(metadata_df.latitude.to_xarray()),
39
+ longitude=(metadata_df.longitude.to_xarray()),
40
+ capacity_kwp=(metadata_df.capacity_kwp.to_xarray()),
41
+ )
33
42
 
34
43
  # Sanity checks, to prevent inf or negative values
35
44
  # Note NaNs are allowed in generation_kw as can have non overlapping time periods for sites
{ocf_data_sampler-0.6.0 → ocf_data_sampler-0.6.1}/ocf_data_sampler.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ocf-data-sampler
3
- Version: 0.6.0
3
+ Version: 0.6.1
4
4
  Author: James Fulton, Peter Dudfield
5
5
  Author-email: Open Climate Fix team <info@openclimatefix.org>
6
6
  License: MIT License