From 2cfd6bb99fe7bea26bc88982911c4de9d7335eb4 Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Thu, 13 Jun 2024 09:31:54 +0200 Subject: [PATCH] amend --- benchmarl/models/gnn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarl/models/gnn.py b/benchmarl/models/gnn.py index cbb16be9..8f535e24 100644 --- 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