Fix jaxtyping syntax error

[a-r-j]

Reference Issues/PRs

What does this implement/fix? Explain your changes

What testing did you do to verify the changes in this PR?

Pull Request Checklist

• Added a note about the modification or contribution to the ./CHANGELOG.md file (if applicable)
• Added appropriate unit test functions in the ./graphein/tests/* directories (if applicable)
• Modify documentation in the corresponding Jupyter Notebook under ./notebooks/ (if applicable)
• Ran python -m py.test tests/ and make sure that all unit tests pass (for small modifications, it might be sufficient to only run the specific test file, e.g., python -m py.test tests/protein/test_graphs.py)
• Checked for style issues by running black . and isort .

[sonarcloud]

Kudos, SonarCloud Quality Gate passed!   

0 Bugs
0 Vulnerabilities
0 Security Hotspots
0 Code Smells

No Coverage information
0.0% Duplication

[codecov-commenter]

Codecov Report

Patch coverage: 40.15% and project coverage change: +3.77 🎉

Comparison is base (8123f42) 40.27% compared to head (088ae02) 44.04%.

📣 This organization is not using Codecov’s GitHub App Integration. We recommend you install it so Codecov can continue to function properly for your repositories. Learn more

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #312      +/-   ##
==========================================
+ Coverage   40.27%   44.04%   +3.77%     
==========================================
  Files          48      113      +65     
  Lines        2811     7765    +4954     
==========================================
+ Hits         1132     3420    +2288     
- Misses       1679     4345    +2666     

Impacted Files	Coverage Δ
graphein/ml/datasets/foldcomp_dataset.py	0.00% <0.00%> (ø)
graphein/ml/diffusion.py	0.00% <0.00%> (ø)
graphein/ml/metrics/__init__.py	0.00% <0.00%> (ø)
graphein/ml/metrics/gdt.py	0.00% <0.00%> (ø)
graphein/ml/metrics/tm_score.py	0.00% <0.00%> (ø)
graphein/ppi/graph_metadata.py	0.00% <0.00%> (ø)
graphein/ppi/visualisation.py	0.00% <0.00%> (ø)
graphein/protein/analysis.py	0.00% <0.00%> (ø)
graphein/protein/features/utils.py	27.77% <0.00%> (ø)
graphein/protein/folding_utils.py	0.00% <0.00%> (ø)
... and 95 more

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