From 2263c97fe8e143566e3a4694f7a22bb6a0f56a8f Mon Sep 17 00:00:00 2001 From: Tibo De Peuter Date: Wed, 1 Apr 2026 17:13:08 +0000 Subject: [PATCH] style: ruff format --- src/MLPs/centralized.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/MLPs/centralized.py b/src/MLPs/centralized.py index 70abbec..14e3a22 100644 --- a/src/MLPs/centralized.py +++ b/src/MLPs/centralized.py @@ -6,6 +6,7 @@ from flax.linen.initializers import constant, orthogonal class Network(nn.Module): hidden_size: int = 256 + @nn.compact def __call__(self, x): # x shape: (batch, obs_dim)