Update template to Lightning 2.0

[johnnynunez]

No description provided.

[codecov-commenter]

Codecov Report

Patch coverage: 100.00% and project coverage change: -0.14 ⚠️

Comparison is base (adc6afe) 83.80% compared to head (4cfeeed) 83.66%.

❗ Current head 4cfeeed differs from pull request most recent head 7778d41. Consider uploading reports for the commit 7778d41 to get more accurate results

Additional details and impacted files

@@            Coverage Diff             @@
##              dev     #548      +/-   ##
==========================================
- Coverage   83.80%   83.66%   -0.14%     
==========================================
  Files           9        9              
  Lines         352      349       -3     
==========================================
- Hits          295      292       -3     
  Misses         57       57              

Impacted Files	Coverage Δ
src/data/mnist_datamodule.py	94.73% <100.00%> (ø)
src/eval.py	90.32% <100.00%> (ø)
src/models/mnist_module.py	98.41% <100.00%> (-0.05%)	⬇️
src/train.py	95.91% <100.00%> (+0.08%)	⬆️
src/utils/pylogger.py	100.00% <100.00%> (ø)
src/utils/rich_utils.py	82.60% <100.00%> (ø)
src/utils/utils.py	59.80% <100.00%> (-0.78%)	⬇️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

☔ View full report in Codecov by Sentry.
📢 Do you have feedback about the report comment? Let us know in this issue.

[johnnynunez]

@ashleve it's important to change pytorch-lightning to lightning

[ashleve]

@johnnynunez Thanks for helping with the update!

[ashleve]

I reverted the pre-commit changes because they were not relevant to updating Lightning and want to address them in a separate pull request.

[ashleve]

WIll also address pytorch compilation in another PR