1
Fork 0

style: ruff format

This commit is contained in:
Tibo De Peuter 2026-04-01 17:13:08 +00:00 committed by cedric
parent 4e3eee8ac6
commit 2263c97fe8

View file

@ -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)