fix formatting issue in nonlinear pde operator

xiangyuan-zhang · Feb 1, 2024 · 89c467c · 89c467c
1 parent 111885b
commit 89c467c
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 14 deletions.
diff --git a/controlgym/envs/allen_cahn.py b/controlgym/envs/allen_cahn.py
@@ -140,8 +140,7 @@ def fourier_nonlinear_op(state_fourier, action):
             )
             right_hand_side = -self.potential_constant * (1 / aa_factor) * np.fft.rfft(
                 aa_state ** 3
-            )[0 : int(self.n_state / 2) + 1]
-            + (np.fft.rfft(self.control_sup, axis=0) @ action)
+            )[0 : int(self.n_state / 2) + 1] + (np.fft.rfft(self.control_sup, axis=0) @ action)
             return right_hand_side
         return fourier_nonlinear_op
 

diff --git a/controlgym/envs/burgers.py b/controlgym/envs/burgers.py
@@ -135,8 +135,7 @@ def fourier_nonlinear_op(state_fourier, action):
                 * self.domain_wavenumbers
                 * (1 / aa_factor)
                 * np.fft.rfft(aa_state**2)[0 : int(self.n_state / 2) + 1]
-            )
-            +(np.fft.rfft(self.control_sup, axis=0) @ action)
+            ) + (np.fft.rfft(self.control_sup, axis=0) @ action)
             return right_hand_side
 
         return fourier_nonlinear_op

diff --git a/controlgym/envs/cahn_hilliard.py b/controlgym/envs/cahn_hilliard.py
@@ -145,8 +145,7 @@ def fourier_nonlinear_op(state_fourier, action):
                 * self.domain_wavenumbers**2
                 * (1 / aa_factor)
                 * np.fft.rfft(aa_state**3)[0 : int(self.n_state / 2) + 1]
-            )
-            +(np.fft.rfft(self.control_sup, axis=0) @ action)
+            ) + (np.fft.rfft(self.control_sup, axis=0) @ action)
             return right_hand_side
 
         return fourier_nonlinear_op

diff --git a/controlgym/envs/fisher.py b/controlgym/envs/fisher.py
@@ -138,8 +138,7 @@ def fourier_nonlinear_op(state_fourier, action):
                 self.reaction_constant
                 * (1 / aa_factor)
                 * np.fft.rfft(aa_state * (1 - aa_state))[0 : int(self.n_state / 2) + 1]
-            )
-            +(np.fft.rfft(self.control_sup, axis=0) @ action)
+            ) + (np.fft.rfft(self.control_sup, axis=0) @ action)
             return right_hand_side
 
         return fourier_nonlinear_op

diff --git a/controlgym/envs/ginzburg_landau.py b/controlgym/envs/ginzburg_landau.py
@@ -131,8 +131,7 @@ def fourier_nonlinear_op(state_fourier, action):
             right_hand_side = (
                 -(1 / aa_factor)
                 * np.fft.rfft(aa_state**3)[0 : int(self.n_state / 2) + 1]
-            )
-            +(np.fft.rfft(self.control_sup, axis=0) @ action)
+            ) + (np.fft.rfft(self.control_sup, axis=0) @ action)
             return right_hand_side
 
         return fourier_nonlinear_op
diff --git a/controlgym/envs/korteweg_de_vries.py b/controlgym/envs/korteweg_de_vries.py
@@ -146,8 +146,7 @@ def fourier_nonlinear_op(state_fourier, action):
                 * self.domain_wavenumbers
                 * (1 / aa_factor)
                 * np.fft.rfft(aa_state**2)[0 : int(self.n_state / 2) + 1]
-            )
-            +(np.fft.rfft(self.control_sup, axis=0) @ action)
+            ) + (np.fft.rfft(self.control_sup, axis=0) @ action)
             return right_hand_side
 
         return fourier_nonlinear_op
diff --git a/controlgym/envs/kuramoto_sivashinsky.py b/controlgym/envs/kuramoto_sivashinsky.py
@@ -135,8 +135,7 @@ def fourier_nonlinear_op(state_fourier, action):
                 * self.domain_wavenumbers
                 * (1 / aa_factor)
                 * np.fft.rfft(aa_state**2)[0 : int(self.n_state / 2) + 1]
-            )
-            +(np.fft.rfft(self.control_sup, axis=0) @ action)
+            ) + (np.fft.rfft(self.control_sup, axis=0) @ action)
             return right_hand_side
 
         return fourier_nonlinear_op