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):
|
class Network(nn.Module):
|
||||||
hidden_size: int = 256
|
hidden_size: int = 256
|
||||||
|
|
||||||
@nn.compact
|
@nn.compact
|
||||||
def __call__(self, x):
|
def __call__(self, x):
|
||||||
# x shape: (batch, obs_dim)
|
# x shape: (batch, obs_dim)
|
||||||
|
|
|
||||||
Reference in a new issue