style: ruff format
This commit is contained in:
parent
4e3eee8ac6
commit
2263c97fe8
1 changed files with 1 additions and 0 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Reference in a new issue