openclimatefix / ocf_datapipes

OCF's DataPipe based dataloader for training and inference
MIT License
13 stars 11 forks source link

Add Threadpool Map to more datapipes for speed ups #181

Closed jacobbieker closed 1 year ago

jacobbieker commented 1 year ago

Pull Request

Description

This adds threadpool mapping to the Irradiance loading. Also adds ability to add threadpools for other parts of the code.

Fixes #180

How Has This Been Tested?

Unit tests

Checklist:

codecov[bot] commented 1 year ago

Codecov Report

Merging #181 (e567b7d) into main (b744c18) will increase coverage by 0.01%. The diff coverage is 83.05%.

:exclamation: Current head e567b7d differs from pull request most recent head cf6a53d. Consider uploading reports for the commit cf6a53d to get more accurate results

@@            Coverage Diff             @@
##             main     #181      +/-   ##
==========================================
+ Coverage   77.85%   77.87%   +0.01%     
==========================================
  Files         123      124       +1     
  Lines        5054     5094      +40     
==========================================
+ Hits         3935     3967      +32     
- Misses       1119     1127       +8     
Impacted Files Coverage Δ
ocf_datapipes/training/pseudo_irradience.py 16.52% <25.00%> (+0.78%) :arrow_up:
ocf_datapipes/utils/parallel.py 85.71% <85.71%> (ø)
..._datapipes/transform/xarray/metnet_preprocessor.py 92.75% <96.66%> (+0.69%) :arrow_up:

:mega: We’re building smart automated test selection to slash your CI/CD build times. Learn more