Refactor training loop from script to class

[sweep-nightly]

Description

This PR refactors the training loop from a script to a class in order to improve code organization and reusability. It introduces a new Trainer class in the src/trainer.py file that encapsulates the training loop logic. The Trainer class takes the model, dataloader, and optimizer as inputs and provides a train method to train the model.

Summary of Changes

• Created a new file src/trainer.py to define the Trainer class.
• Modified src/main.py to import and use the Trainer class.
• Replaced the existing training loop in src/main.py with a call to the train method of the Trainer instance.

Fixes #6.

---

🎉 Latest improvements to Sweep:

• Sweep can now passively improve your repository! Check out Rules to learn more.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

[sweep-nightly]

Sandbox Executions

• Check src/trainer.py ✓

Sandbox logs for https://github.com/sweepai/evals/commit/198424f4ca8d266a11db1d9ff527bddf86e0b106

trunk fmt src/trainer.py || exit 0 1/4 ✓

 ✔ Formatted src/trainer.py
Re-checking autofixed files...


Checked 1 file
✔ No issues
Run trunk upgrade to upgrade 1 linter

trunk check --fix --print-failures src/trainer.py 2/4 ✓

 ✔ Auto-fixed src/trainer.py
Re-checking autofixed files...


Checked 1 file
✔ No issues

if [[ $(echo "src/trainer.py" | grep 'test.*\.py$') ]]; then PYTHONPATH=. python src/trainer.py; else exit 0; fi 3/4 ✓

(nothing was outputted)

if [[ "src/trainer.py" == *test*.py ]]; then PYTHONPATH=. pytest src/trainer.py; else exit 0; fi 4/4 ✓

(nothing was outputted)

• Check src/main.py ✓

Sandbox logs for https://github.com/sweepai/evals/commit/8d9efe394a4d15351d0dc9fef61cfd12d62bb386

trunk fmt src/main.py || exit 0 1/4 ✓

 ✔ Formatted src/main.py
Re-checking autofixed files...

 ✔ Formatted src/main.py
Re-checking autofixed files...


Checked 1 file
✔ No issues
Run trunk upgrade to upgrade 1 linter

trunk check --fix --print-failures src/main.py 2/4 ✓

 ✔ Auto-fixed src/main.py
Re-checking autofixed files...


  ISSUES  
src/main.py:1:17
 1:17  high  `numpy` imported but unused  ruff/F401
Checked 1 file
1 existing issue (1 auto-fixable)

if [[ $(echo "src/main.py" | grep 'test.*\.py$') ]]; then PYTHONPATH=. python src/main.py; else exit 0; fi 3/4 ✓

(nothing was outputted)

if [[ "src/main.py" == *test*.py ]]; then PYTHONPATH=. pytest src/main.py; else exit 0; fi 4/4 ✓

(nothing was outputted)

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: All docstrings and comments should be up to date.
• Apply: Code should be properly formatted and indented.
• Apply: Variable and function names should be descriptive and follow a consistent naming convention.
• Apply: Imports should be organized and grouped together.
• Apply: There should be no unused imports or variables.
• Apply: Code should be properly commented and include docstrings for functions and classes.