Merge branch 'main' into inv_sqr

a-r-j · Feb 7, 2024 · 95da831 · 95da831
2 parents ef254bf + 897e091
commit 95da831
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,8 +1,13 @@
 ### 0.2.6 (UNRELEASED)
 
+### Models
+
+* Adds missing `pos` attribute to GearNet `required_batch_attributes` (fixes [#73](https://github.com/a-r-j/ProteinWorkshop/issues/73)) [#74](https://github.com/a-r-j/ProteinWorkshop/pull/74)
+
 ### Framework
 
-* Adds InverseSquareRoot LR scheduler [#71](https://github.com/a-r-j/ProteinWorkshop/pull/71)
+* Adds `InverseSquareRoot` LR scheduler [#71](https://github.com/a-r-j/ProteinWorkshop/pull/71)
+
 
 ### 0.2.5 (28/12/2023)
 

diff --git a/proteinworkshop/models/graph_encoders/gear_net.py b/proteinworkshop/models/graph_encoders/gear_net.py
@@ -125,6 +125,7 @@ def required_batch_attributes(self) -> Set[str]:
         """
         return {
             "x",
+            "pos",
             "edge_index",
             "edge_type",
             "edge_attr",