Add end-to-end training test

[dfulu]

Pull Request

Description

Add a test to test training the model using pytorch lightning end-to-end. This adds test coverage to a wider range of the functions in the library. Particularly the pytorch lightning integration functions around computing metrics for training and validation.

Closes #109

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

All modified and coverable lines are covered by tests ✅

Comparison is base (b341182) 50.85% compared to head (eed5241) 57.67%.
Report is 2 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #113      +/-   ##
==========================================
+ Coverage   50.85%   57.67%   +6.81%     
==========================================
  Files          26       26              
  Lines        1701     1701              
==========================================
+ Hits          865      981     +116     
+ Misses        836      720     -116     

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