Add Pyright type checking

ml-energy / zeus

Deep Learning Energy Measurement and Optimization

https://ml.energy/zeus

Apache License 2.0

179 stars 24 forks source link

Add Pyright type checking #58

Closed jaywonchung closed 2 months ago

jaywonchung commented 2 months ago

This PR adds pyright type checking across the entire package.

@show981111 Could you look over the changes in zeus/optimizer/batch_size? I made good efforts to maintain functionality and add checks that semantically make sense. All tests pass (locally) and there are no type errors originating from the BSO now.

TODOs

[x] Merge in changes in gpu.py
[x] Make CI pass again

jaywonchung commented 2 months ago

@show981111 Ready for review!