Add support for PyG 2.4+

[a-r-j]

Reference Issues/PRs

What does this implement/fix? Explain your changes

2.4 introduced breaking changes to Data.keys and Batch.keys, replacing the attribute with a method (Data.keys()). This PR adds looseversion to select the keys based on the syntax allowed by old and new PyG versions.

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[bot]]

Kudos, SonarCloud Quality Gate passed!   

0 Bugs
0 Vulnerabilities
0 Security Hotspots
0 Code Smells

No Coverage information
0.0% Duplication