Fix `sequence_edges` when b is a Data object

a-r-j / ProteinWorkshop

Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities. (ICLR 2024)

https://proteins.sh/

MIT License

194 stars 16 forks source link

Fix `sequence_edges` when b is a Data object #80

Closed linusyh closed 7 months ago

linusyh commented 7 months ago

proteinworkshop.features.edges.sequence_edges currently does not work when argument b is a PyG Data object. This presents a simple fix.

amorehead commented 7 months ago

LGTM. Thanks for the PR, @linusyh!