diff --git a/benchmarl/models/gnn.py b/benchmarl/models/gnn.py index 855ba81b..c238b17a 100644 --- a/benchmarl/models/gnn.py +++ b/benchmarl/models/gnn.py @@ -466,6 +466,7 @@ class GnnConfig(ModelConfig): """Dataclass config for a :class:`~benchmarl.models.Gnn`.""" topology: str = MISSING + self_loops: bool = MISSING gnn_class: Type[torch_geometric.nn.MessagePassing] = MISSING