Add type hints

[james77777778]

Fix #21

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (a50003c) 80.82% compared to head (88e2684) 80.93%.

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #25      +/-   ##
==========================================
+ Coverage   80.82%   80.93%   +0.10%     
==========================================
  Files          24       24              
  Lines        2555     2559       +4     
==========================================
+ Hits         2065     2071       +6     
+ Misses        490      488       -2     

Flag	Coverage Δ
kimm	80.93% <100.00%> (+0.10%)	⬆️
kimm-jax	?
kimm-numpy	80.93% <100.00%> (+0.10%)	⬆️
kimm-tensorflow	80.93% <100.00%> (+0.10%)	⬆️
kimm-torch	80.93% <100.00%> (+0.10%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

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