feat: nequip model in batch

[njzjz]

No description provided.

[pep8speaks]

Hello @njzjz! Thanks for opening this PR. We checked the lines you've touched for PEP 8 issues, and found:

• In the file deepmd_gnn/nequip.py:

Line 555:80: E501 line too long (84 > 79 characters)
Line 563:80: E501 line too long (84 > 79 characters)
Line 585:80: E501 line too long (86 > 79 characters)

[codecov]

Codecov Report

Attention: Patch coverage is 90.69767% with 4 lines in your changes missing coverage. Please review.

Project coverage is 90.24%. Comparing base (f4c6b22) to head (87f687d).
Report is 1 commits behind head on master.

Files with missing lines	Patch %	Lines
deepmd_gnn/nequip.py	90.69%	4 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##           master      #28      +/-   ##
==========================================
- Coverage   90.46%   90.24%   -0.22%     
==========================================
  Files           5        5              
  Lines         493      482      -11     
==========================================
- Hits          446      435      -11     
  Misses         47       47              

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