benchmarl.models.GnnConfig

class GnnConfig(topology: str, self_loops: bool, gnn_class: Type[torch_geometric.nn.MessagePassing], gnn_kwargs: dict | None = None, position_key: str | None = None, pos_features: int | None = 0, velocity_key: str | None = None, vel_features: int | None = 0, exclude_pos_from_node_features: bool | None = None, edge_radius: float | None = None)

Bases: ModelConfig

Dataclass config for a Gnn.

static associated_class()

The associated Model class