benchmarl.models.GnnConfig

class GnnConfig(topology: str, self_loops: bool, gnn_class: Type[torch_geometric.nn.MessagePassing], gnn_kwargs: dict | None = None)

Bases: ModelConfig

Dataclass config for a Gnn.

static associated_class()

The associated Model class