amend

facebookresearch · Sep 3, 2024 · 363b728 · 363b728
1 parent fd1131a
commit 363b728
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/benchmarl/models/gnn.py b/benchmarl/models/gnn.py
@@ -458,14 +458,13 @@ class GnnConfig(ModelConfig):
     topology: str = MISSING
     self_loops: bool = MISSING
 
-    pos_features: int = MISSING
-    vel_features: int = MISSING
-
     gnn_class: Type[torch_geometric.nn.MessagePassing] = MISSING
     gnn_kwargs: Optional[dict] = None
 
     position_key: Optional[str] = None
+    pos_features: Optional[int] = 0
     velocity_key: Optional[str] = None
+    vel_features: Optional[int] = 0
     exclude_pos_from_node_features: Optional[bool] = None
     edge_radius: Optional[float] = None