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:

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

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

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

```diff @@ 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 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.