added pyg mpnn layer

lucidrains · Feb 27, 2021 · f56d616 · f56d616
1 parent 63d8997
commit f56d616
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/.DS_Store b/.DS_Store
diff --git a/.gitignore b/.gitignore
@@ -6,6 +6,10 @@ __pycache__/
 # C extensions
 *.so
 
+# mac files
+*.DS_Store
+.DS_Store
+
 # Distribution / packaging
 .Python
 build/

diff --git a/README.md b/README.md
@@ -46,7 +46,7 @@ feats, coors = layer2(feats, coors, edges) # (1, 16, 512), (1, 16, 3)
 ## Todo
 
 - [ ] masking
-- [ ] add integration with pytorch geometric
+- [x] add integration with pytorch geometric
 - [x] add tests for se3 equivariance
 - [x] add an EGAT (attention flavored variant)