Ignore formating config sections

Toni-SM · Nov 4, 2024 · 6942e7c · 6942e7c
1 parent 6ef70fe
commit 6942e7c
Show file tree

Hide file tree

Showing 38 changed files with 76 additions and 0 deletions.
diff --git a/skrl/agents/jax/a2c/a2c.py b/skrl/agents/jax/a2c/a2c.py
@@ -16,6 +16,7 @@
 from skrl.resources.schedulers.jax import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-jax]
 A2C_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -56,6 +57,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 def compute_gae(rewards: np.ndarray,

diff --git a/skrl/agents/jax/cem/cem.py b/skrl/agents/jax/cem/cem.py
@@ -15,6 +15,7 @@
 from skrl.resources.optimizers.jax import Adam
 
 
+# fmt: off
 # [start-config-dict-jax]
 CEM_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -47,6 +48,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 class CEM(Agent):

diff --git a/skrl/agents/jax/ddpg/ddpg.py b/skrl/agents/jax/ddpg/ddpg.py
@@ -15,6 +15,7 @@
 from skrl.resources.optimizers.jax import Adam
 
 
+# fmt: off
 # [start-config-dict-jax]
 DDPG_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -58,6 +59,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 # https://jax.readthedocs.io/en/latest/faq.html#strategy-1-jit-compiled-helper-function

diff --git a/skrl/agents/jax/dqn/ddqn.py b/skrl/agents/jax/dqn/ddqn.py
@@ -15,6 +15,7 @@
 from skrl.resources.optimizers.jax import Adam
 
 
+# fmt: off
 # [start-config-dict-jax]
 DDQN_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -57,6 +58,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 # https://jax.readthedocs.io/en/latest/faq.html#strategy-1-jit-compiled-helper-function

diff --git a/skrl/agents/jax/dqn/dqn.py b/skrl/agents/jax/dqn/dqn.py
@@ -15,6 +15,7 @@
 from skrl.resources.optimizers.jax import Adam
 
 
+# fmt: off
 # [start-config-dict-jax]
 DQN_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -57,6 +58,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 # https://jax.readthedocs.io/en/latest/faq.html#strategy-1-jit-compiled-helper-function

diff --git a/skrl/agents/jax/ppo/ppo.py b/skrl/agents/jax/ppo/ppo.py
@@ -16,6 +16,7 @@
 from skrl.resources.schedulers.jax import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-jax]
 PPO_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -63,6 +64,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 def compute_gae(rewards: np.ndarray,

diff --git a/skrl/agents/jax/rpo/rpo.py b/skrl/agents/jax/rpo/rpo.py
@@ -16,6 +16,7 @@
 from skrl.resources.schedulers.jax import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-jax]
 RPO_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -64,6 +65,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 def compute_gae(rewards: np.ndarray,

diff --git a/skrl/agents/jax/sac/sac.py b/skrl/agents/jax/sac/sac.py
@@ -16,6 +16,7 @@
 from skrl.resources.optimizers.jax import Adam
 
 
+# fmt: off
 # [start-config-dict-jax]
 SAC_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -57,6 +58,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 # https://jax.readthedocs.io/en/latest/faq.html#strategy-1-jit-compiled-helper-function

diff --git a/skrl/agents/jax/td3/td3.py b/skrl/agents/jax/td3/td3.py
@@ -15,6 +15,7 @@
 from skrl.resources.optimizers.jax import Adam
 
 
+# fmt: off
 # [start-config-dict-jax]
 TD3_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -62,6 +63,7 @@
     }
 }
 # [end-config-dict-jax]
+# fmt: on
 
 
 # https://jax.readthedocs.io/en/latest/faq.html#strategy-1-jit-compiled-helper-function

diff --git a/skrl/agents/torch/a2c/a2c.py b/skrl/agents/torch/a2c/a2c.py
@@ -15,6 +15,7 @@
 from skrl.resources.schedulers.torch import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-torch]
 A2C_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -55,6 +56,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class A2C(Agent):

diff --git a/skrl/agents/torch/a2c/a2c_rnn.py b/skrl/agents/torch/a2c/a2c_rnn.py
@@ -15,6 +15,7 @@
 from skrl.resources.schedulers.torch import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-torch]
 A2C_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -55,6 +56,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class A2C_RNN(Agent):

diff --git a/skrl/agents/torch/amp/amp.py b/skrl/agents/torch/amp/amp.py
@@ -15,6 +15,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 AMP_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -72,6 +73,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class AMP(Agent):

diff --git a/skrl/agents/torch/cem/cem.py b/skrl/agents/torch/cem/cem.py
@@ -12,6 +12,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 CEM_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -44,6 +45,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class CEM(Agent):

diff --git a/skrl/agents/torch/ddpg/ddpg.py b/skrl/agents/torch/ddpg/ddpg.py
@@ -13,6 +13,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 DDPG_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -56,6 +57,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class DDPG(Agent):

diff --git a/skrl/agents/torch/ddpg/ddpg_rnn.py b/skrl/agents/torch/ddpg/ddpg_rnn.py
@@ -13,6 +13,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 DDPG_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -56,6 +57,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class DDPG_RNN(Agent):

diff --git a/skrl/agents/torch/dqn/ddqn.py b/skrl/agents/torch/dqn/ddqn.py
@@ -13,6 +13,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 DDQN_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -55,6 +56,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class DDQN(Agent):

diff --git a/skrl/agents/torch/dqn/dqn.py b/skrl/agents/torch/dqn/dqn.py
@@ -13,6 +13,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 DQN_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -55,6 +56,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class DQN(Agent):

diff --git a/skrl/agents/torch/ppo/ppo.py b/skrl/agents/torch/ppo/ppo.py
@@ -15,6 +15,7 @@
 from skrl.resources.schedulers.torch import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-torch]
 PPO_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -64,6 +65,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class PPO(Agent):

diff --git a/skrl/agents/torch/ppo/ppo_rnn.py b/skrl/agents/torch/ppo/ppo_rnn.py
@@ -15,6 +15,7 @@
 from skrl.resources.schedulers.torch import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-torch]
 PPO_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -62,6 +63,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class PPO_RNN(Agent):

diff --git a/skrl/agents/torch/q_learning/q_learning.py b/skrl/agents/torch/q_learning/q_learning.py
@@ -10,6 +10,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 Q_LEARNING_DEFAULT_CONFIG = {
     "discount_factor": 0.99,        # discount factor (gamma)
@@ -34,6 +35,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class Q_LEARNING(Agent):

diff --git a/skrl/agents/torch/rpo/rpo.py b/skrl/agents/torch/rpo/rpo.py
@@ -15,6 +15,7 @@
 from skrl.resources.schedulers.torch import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-torch]
 RPO_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -63,6 +64,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class RPO(Agent):

diff --git a/skrl/agents/torch/rpo/rpo_rnn.py b/skrl/agents/torch/rpo/rpo_rnn.py
@@ -15,6 +15,7 @@
 from skrl.resources.schedulers.torch import KLAdaptiveLR
 
 
+# fmt: off
 # [start-config-dict-torch]
 RPO_DEFAULT_CONFIG = {
     "rollouts": 16,                 # number of rollouts before updating
@@ -63,6 +64,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class RPO_RNN(Agent):

diff --git a/skrl/agents/torch/sac/sac.py b/skrl/agents/torch/sac/sac.py
@@ -15,6 +15,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 SAC_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -56,6 +57,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class SAC(Agent):

diff --git a/skrl/agents/torch/sac/sac_rnn.py b/skrl/agents/torch/sac/sac_rnn.py
@@ -15,6 +15,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 SAC_DEFAULT_CONFIG = {
     "gradient_steps": 1,            # gradient steps
@@ -56,6 +57,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class SAC_RNN(Agent):

diff --git a/skrl/agents/torch/sarsa/sarsa.py b/skrl/agents/torch/sarsa/sarsa.py
@@ -10,6 +10,7 @@
 from skrl.models.torch import Model
 
 
+# fmt: off
 # [start-config-dict-torch]
 SARSA_DEFAULT_CONFIG = {
     "discount_factor": 0.99,        # discount factor (gamma)
@@ -34,6 +35,7 @@
     }
 }
 # [end-config-dict-torch]
+# fmt: on
 
 
 class SARSA(Agent):