Sparse changes to allow for non-eager 'None' sizes

danielegrattarola / spektral

Graph Neural Networks with Keras and Tensorflow 2.

https://graphneural.network

MIT License

2.37k stars 334 forks source link

Sparse changes to allow for non-eager 'None' sizes #422

Open dmadisetti opened 1 year ago

dmadisetti commented 1 year ago

Allows for GAT to be used with tensors of indeterminate size in single mode

danielegrattarola commented 1 year ago

Hey, thanks for the PR! This is currently not passing the tests, I'll try to have a look into it to see if it can be fixed.

dmadisetti commented 1 year ago

Weird. Not sure why this isn't failing locally. Looks like explicitly setting the signture type could work? I changed tf versions and will submit a fix

dmadisetti commented 1 year ago

Works!