add if statement

[peterdudfield]

Pull Request

Description

Fix tests

Fixes #

How Has This Been Tested?

CI tests

• Yes

Checklist:

• My code follows OCF's coding style guidelines
• I have performed a self-review of my own code
• I have made corresponding changes to the documentation
• I have added tests that prove my fix is effective or that my feature works
• I have checked my code and corrected any misspellings

[codecov]

Codecov Report

Attention: Patch coverage is 33.33333% with 2 lines in your changes are missing coverage. Please review.

Project coverage is 76.51%. Comparing base (e85b5a1) to head (282677f).
Report is 19 commits behind head on main.

Files	Patch %	Lines
ocf_datapipes/training/windnet.py	33.33%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #295      +/-   ##
==========================================
- Coverage   76.52%   76.51%   -0.02%     
==========================================
  Files         124      125       +1     
  Lines        5875     5909      +34     
==========================================
+ Hits         4496     4521      +25     
- Misses       1379     1388       +9     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.