amend

facebookresearch · Jun 13, 2024 · 2cfd6bb · 2cfd6bb
1 parent 2eac93c
commit 2cfd6bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarl/models/gnn.py b/benchmarl/models/gnn.py
@@ -418,10 +418,10 @@ class GnnConfig(ModelConfig):
     self_loops: bool = MISSING
 
     gnn_class: Type[torch_geometric.nn.MessagePassing] = MISSING
-    gnn_kwargs: Optional[dict] = None
+    exclude_pos_from_node_features: bool = MISSING
 
+    gnn_kwargs: Optional[dict] = None
     position_key: Optional[str] = None
-    exclude_pos_from_node_features: bool = MISSING
     velocity_key: Optional[str] = None
     edge_radius: Optional[float] = None