fix: fix ruff

PKU-Alignment · May 25, 2023 · 538b14a · 538b14a
1 parent 0da3326
commit 538b14a
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/omnisafe/models/actor/perturbation_actor.py b/omnisafe/models/actor/perturbation_actor.py
@@ -56,7 +56,7 @@ def __init__(  # pylint: disable=too-many-arguments
 
         self.vae = VAE(obs_space, act_space, hidden_sizes, activation, weight_initialization_mode)
         self.perturbation = build_mlp_network(
-            sizes=[self._obs_dim + self._act_dim, *hidden_sizes] + [self._act_dim],
+            sizes=[self._obs_dim + self._act_dim, *hidden_sizes, self._act_dim],
             activation=activation,
             output_activation='tanh',
             weight_initialization_mode=weight_initialization_mode,

diff --git a/omnisafe/models/actor/vae_actor.py b/omnisafe/models/actor/vae_actor.py
@@ -53,12 +53,12 @@ def __init__(  # pylint: disable=too-many-arguments
         self._latent_dim = self._act_dim * 2
 
         self._encoder = build_mlp_network(
-            sizes=[self._obs_dim + self._act_dim, *hidden_sizes] + [self._latent_dim * 2],
+            sizes=[self._obs_dim + self._act_dim, *hidden_sizes, self._latent_dim * 2],
             activation=activation,
             weight_initialization_mode=weight_initialization_mode,
         )
         self._decoder = build_mlp_network(
-            sizes=[self._obs_dim + self._latent_dim, *hidden_sizes] + [self._act_dim],
+            sizes=[self._obs_dim + self._latent_dim, *hidden_sizes, self._act_dim],
             activation=activation,
             weight_initialization_mode=weight_initialization_mode,
         )

diff --git a/omnisafe/models/critic/q_critic.py b/omnisafe/models/critic/q_critic.py
@@ -97,7 +97,7 @@ def __init__(
                 critic = nn.Sequential(obs_encoder, net)
             else:
                 net = build_mlp_network(
-                    [self._obs_dim + self._act_dim, *hidden_sizes] + [1],
+                    [self._obs_dim + self._act_dim, *hidden_sizes, 1],
                     activation=activation,
                     weight_initialization_mode=weight_initialization_mode,
                 )

diff --git a/omnisafe/models/offline/dice.py b/omnisafe/models/offline/dice.py
@@ -69,7 +69,7 @@ def __init__(
         self.activation = activation
         self.hidden_sizes = hidden_sizes
         self.net = build_mlp_network(
-            [self._obs_dim, *list(hidden_sizes)] + [self._out_dim],
+            [self._obs_dim, *list(hidden_sizes), self._out_dim],
             activation=activation,
             weight_initialization_mode=weight_initialization_mode,
         )