Codacy formatting

Author: vl-dud

deepxde/model.py

@@ -109,9 +109,12 @@ def compile(
 
                 - For backend JAX:
 
-                    - `linear_schedule <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.linear_schedule>`_: ("linear", end_value, transition_steps)
-                    - `cosine_decay_schedule <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.cosine_decay_schedule>`_: ("cosine", decay_steps, alpha)
-                    - `exponential_decay <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.exponential_decay>`_: ("exponential", transition_steps, decay_rate)
+                    - `linear_schedule <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.linear_schedule>`_:
+                      ("linear", end_value, transition_steps)
+                    - `cosine_decay_schedule <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.cosine_decay_schedule>`_:
+                      ("cosine", decay_steps, alpha)
+                    - `exponential_decay <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.exponential_decay>`_:
+                      ("exponential", transition_steps, decay_rate)
 
             loss_weights: A list specifying scalar coefficients (Python floats) to
                 weight the loss contributions. The loss value that will be minimized by

deepxde/optimizers/jax/optimizers.py

@@ -38,9 +38,9 @@ def _get_learningrate(lr, decay):
         return lr
     if decay[0] == "linear":
         return optax.linear_schedule(lr, decay[1], decay[2])
-    elif decay[0] == "cosine":
+    if decay[0] == "cosine":
         return optax.cosine_decay_schedule(lr, decay[1], decay[2])
-    elif decay[0] == "exponential":
+    if decay[0] == "exponential":
         return optax.exponential_decay(lr, decay[1], decay[2])
 
     raise NotImplementedError(