Running under black and isort

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Mar 6, 2024 · b8199f2 · b8199f2
1 parent b825bbe
commit b8199f2
Show file tree

Hide file tree

Showing 82 changed files with 539 additions and 503 deletions.
diff --git a/examples/DeepONet/miscellaneous/notebooks/timeint_DeepONet_simple_manufactured_pinn.py b/examples/DeepONet/miscellaneous/notebooks/timeint_DeepONet_simple_manufactured_pinn.py
@@ -23,16 +23,17 @@
 
 
 import os
+
 import matplotlib.pyplot as plt
 import numpy as np
 
+from simulai.file import SPFile
+from simulai.optimization import Optimizer
+from simulai.residuals import SymbolicOperator
 
 # In[3]:
 
 
-from simulai.file import SPFile
-from simulai.optimization import Optimizer
-from simulai.residuals import SymbolicOperator
 
 
 # #### Basic configuration
@@ -281,7 +282,6 @@ def model():
 
 from scipy.integrate import odeint
 
-
 # #### Pendulum numerical solver
 
 # In[ ]:

diff --git a/examples/DeepONet/miscellaneous/scripts/polynomial_integration.py b/examples/DeepONet/miscellaneous/scripts/polynomial_integration.py
@@ -1,9 +1,9 @@
 import numpy as np
 import torch
 
+from simulai.optimization import Optimizer
 from simulai.residuals import SymbolicOperator
 from simulai.tokens import Dot, Gp
-from simulai.optimization import Optimizer
 
 
 def model():

diff --git a/examples/ESN-RC/scripts/independent_esn_rc_nonlinear_forcing.py b/examples/ESN-RC/scripts/independent_esn_rc_nonlinear_forcing.py
@@ -19,10 +19,8 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from examples.utils.oscillator_solver import (
-    oscillator_solver,
-    oscillator_solver_forcing,
-)
+from examples.utils.oscillator_solver import (oscillator_solver,
+                                              oscillator_solver_forcing)
 from simulai.models import ModelPool
 from simulai.regression import EchoStateNetwork
 from simulai.utilities import make_temp_directory

diff --git a/examples/MapValid/map_valid_reshaper.py b/examples/MapValid/map_valid_reshaper.py
@@ -23,11 +23,8 @@
 from simulai.io import BatchCopy, MapValid
 from simulai.metrics import L2Norm, MemorySizeEval
 from simulai.models import ModelPool
-from simulai.normalization import (
-    BatchNormalization,
-    UnitaryNormalization,
-    UnitarySymmetricalNormalization,
-)
+from simulai.normalization import (BatchNormalization, UnitaryNormalization,
+                                   UnitarySymmetricalNormalization)
 from simulai.rom import IPOD
 from simulai.simulation import Pipeline
 

diff --git a/examples/OpInf/scripts/lorenz_96_chaotic.py b/examples/OpInf/scripts/lorenz_96_chaotic.py
@@ -13,7 +13,6 @@
 #     limitations under the License.
 
 import os
-
 #!/usr/bin/env python
 import warnings
 

diff --git a/examples/OpInf/scripts/lorenz_96_chaotic_multiple.py b/examples/OpInf/scripts/lorenz_96_chaotic_multiple.py
@@ -14,7 +14,6 @@
 
 import os
 import time
-
 #!/usr/bin/env python
 import warnings
 

diff --git a/examples/OpInf/scripts/lorenz_96_stability.py b/examples/OpInf/scripts/lorenz_96_stability.py
@@ -14,7 +14,6 @@
 
 import os
 import pickle
-
 #!/usr/bin/env python
 import warnings
 from argparse import ArgumentParser

diff --git a/examples/OpInf/scripts/opinf_nonlinear_network.py b/examples/OpInf/scripts/opinf_nonlinear_network.py
@@ -25,12 +25,8 @@
 from examples.utils.lorenz_solver import lorenz_solver
 from simulai.math.integration import LSODA, ClassWrapper
 from simulai.optimization import Optimizer, ScipyInterface
-from simulai.regression import (
-    AutoEncoderKoopman,
-    DenseNetwork,
-    KoopmanNetwork,
-    OpInfNetwork,
-)
+from simulai.regression import (AutoEncoderKoopman, DenseNetwork,
+                                KoopmanNetwork, OpInfNetwork)
 
 
 class LorenzJacobian:

diff --git a/examples/OpInf/scripts/svd_lorenz_96_chaotic.py b/examples/OpInf/scripts/svd_lorenz_96_chaotic.py
@@ -13,7 +13,6 @@
 #     limitations under the License.
 
 import os
-
 #!/usr/bin/env python
 import warnings
 

diff --git a/examples/PINN/scripts/Bioreactor/001-Bioreactor_ODEs.py b/examples/PINN/scripts/Bioreactor/001-Bioreactor_ODEs.py
@@ -7,9 +7,9 @@
 """
 
 """ Python Libraries"""
-import pandas as pd
-import numpy as np
 import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
 from scipy.integrate import solve_ivp
 
 """ Kinetics constants"""

diff --git a/examples/PINN/scripts/Bioreactor/002-Bioreactor_PINN.py b/examples/PINN/scripts/Bioreactor/002-Bioreactor_PINN.py
@@ -23,24 +23,25 @@
 """
 
 """    Import Python Libraries    """
+import math
 import os
+import random
 from argparse import ArgumentParser
 from typing import List
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
-import math
-import random
 import torch
 
 torch.set_default_dtype(torch.float64)
 torch.set_default_tensor_type(torch.DoubleTensor)
 
 from simulai import ARRAY_DTYPE
+from simulai.file import SPFile
 from simulai.optimization import Optimizer, PIRMSELoss, ScipyInterface
 from simulai.residuals import SymbolicOperator
 from simulai.templates import NetworkTemplate, guarantee_device
-from simulai.file import SPFile
 
 """#########################################################################"""
 """REPRODUCIBILITY                                                          """
@@ -301,8 +302,8 @@ def adaptative_scale_factors(alpha: float = None):
     activations_funct = "tanh"
 
     def model():
-        from simulai.regression import SLFNN, ConvexDenseNetwork
         from simulai.models import ImprovedDenseNetwork
+        from simulai.regression import SLFNN, ConvexDenseNetwork
 
         # Configuration for the fully-connected network
         config = {
@@ -370,10 +371,11 @@ def model_():
 
         torch.set_default_dtype(torch.float64)
 
-        from simulai.templates import NetworkTemplate, guarantee_device
         import numpy as np
+
         from simulai.models import ImprovedDenseNetwork
         from simulai.regression import SLFNN, ConvexDenseNetwork
+        from simulai.templates import NetworkTemplate, guarantee_device
 
         depth = 3
         width = 64
@@ -382,8 +384,8 @@ def model_():
 
         # Model used for initialization
         def sub_model():
-            from simulai.regression import SLFNN, ConvexDenseNetwork
             from simulai.models import ImprovedDenseNetwork
+            from simulai.regression import SLFNN, ConvexDenseNetwork
 
             # Configuration for the fully-connected network
             config = {

diff --git a/examples/PINN/scripts/Bioreactor_Multifidelity.py b/examples/PINN/scripts/Bioreactor_Multifidelity.py
@@ -23,20 +23,21 @@
 """
 
 """    Import Python Libraries    """
+import random
 from argparse import ArgumentParser
 from typing import List
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
-import random
 import torch
 
 torch.set_default_dtype(torch.float64)
 
+from simulai.file import SPFile
 from simulai.optimization import Optimizer, PIRMSELoss, ScipyInterface
 from simulai.residuals import SymbolicOperator
 from simulai.templates import NetworkTemplate, guarantee_device
-from simulai.file import SPFile
 
 """    Variables    """
 # Bioreactor
@@ -120,8 +121,8 @@ def Epoch_Decay(iteration_number):
     activations_funct = "tanh"
 
     def model():
-        from simulai.regression import SLFNN, ConvexDenseNetwork
         from simulai.models import ImprovedDenseNetwork
+        from simulai.regression import SLFNN, ConvexDenseNetwork
 
         scale_factors = np.array([1, 1, 1, 1])
 
@@ -183,10 +184,11 @@ def model_():
 
         torch.set_default_dtype(torch.float64)
 
-        from simulai.templates import NetworkTemplate, guarantee_device
         import numpy as np
+
         from simulai.models import ImprovedDenseNetwork
         from simulai.regression import SLFNN, ConvexDenseNetwork
+        from simulai.templates import NetworkTemplate, guarantee_device
 
         t_max = 72.0
         n_intervals = 72
@@ -198,8 +200,8 @@ def model_():
 
         # Model used for initialization
         def sub_model():
-            from simulai.regression import SLFNN, ConvexDenseNetwork
             from simulai.models import ImprovedDenseNetwork
+            from simulai.regression import SLFNN, ConvexDenseNetwork
 
             scale_factors = np.array([1, 1, 1, 1])
 

diff --git a/examples/PINN/scripts/Bioreactor_Multifidelity_AutoTimeStep.py b/examples/PINN/scripts/Bioreactor_Multifidelity_AutoTimeStep.py
@@ -23,20 +23,21 @@
 """
 
 """    Import Python Libraries    """
+import random
 from argparse import ArgumentParser
 from typing import List
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
-import random
 import torch
 
 torch.set_default_dtype(torch.float64)
 
+from simulai.file import SPFile
 from simulai.optimization import Optimizer, PIRMSELoss, ScipyInterface
 from simulai.residuals import SymbolicOperator
 from simulai.templates import NetworkTemplate, guarantee_device
-from simulai.file import SPFile
 
 """    Variables    """
 # Bioreactor
@@ -147,8 +148,8 @@ def Delta_t(i, last_delta_t):
     activations_funct = "tanh"
 
     def model():
-        from simulai.regression import SLFNN, ConvexDenseNetwork
         from simulai.models import ImprovedDenseNetwork
+        from simulai.regression import SLFNN, ConvexDenseNetwork
 
         scale_factors = np.array([1, 1, 1, 1])
 
@@ -210,10 +211,11 @@ def model_():
 
         torch.set_default_dtype(torch.float64)
 
-        from simulai.templates import NetworkTemplate, guarantee_device
         import numpy as np
+
         from simulai.models import ImprovedDenseNetwork
         from simulai.regression import SLFNN, ConvexDenseNetwork
+        from simulai.templates import NetworkTemplate, guarantee_device
 
         depth = 3
         width = 50
@@ -222,8 +224,8 @@ def model_():
 
         # Model used for initialization
         def sub_model():
-            from simulai.regression import SLFNN, ConvexDenseNetwork
             from simulai.models import ImprovedDenseNetwork
+            from simulai.regression import SLFNN, ConvexDenseNetwork
 
             scale_factors = np.array([1, 1, 1, 1])
 

diff --git a/examples/PINN/scripts/Bioreactor_Multifidelity_AutoTimeStep_mod.py b/examples/PINN/scripts/Bioreactor_Multifidelity_AutoTimeStep_mod.py
@@ -23,21 +23,22 @@
 """
 
 """    Import Python Libraries    """
+import random
 from argparse import ArgumentParser
 from typing import List
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
-import random
 import torch
 
 torch.set_default_dtype(torch.float64)
 
 from simulai import ARRAY_DTYPE
+from simulai.file import SPFile
 from simulai.optimization import Optimizer, PIRMSELoss, ScipyInterface
 from simulai.residuals import SymbolicOperator
 from simulai.templates import NetworkTemplate, guarantee_device
-from simulai.file import SPFile
 
 """    Variables    """
 # Bioreactor
@@ -148,8 +149,8 @@ def Delta_t(i, last_delta_t):
     activations_funct = "tanh"
 
     def model():
-        from simulai.regression import SLFNN, ConvexDenseNetwork
         from simulai.models import ImprovedDenseNetwork
+        from simulai.regression import SLFNN, ConvexDenseNetwork
 
         # Configuration for the fully-connected network
         config = {
@@ -217,10 +218,11 @@ def model_():
 
         torch.set_default_dtype(torch.float64)
 
-        from simulai.templates import NetworkTemplate, guarantee_device
         import numpy as np
+
         from simulai.models import ImprovedDenseNetwork
         from simulai.regression import SLFNN, ConvexDenseNetwork
+        from simulai.templates import NetworkTemplate, guarantee_device
 
         depth = 3
         width = 50
@@ -229,8 +231,8 @@ def model_():
 
         # Model used for initialization
         def sub_model():
-            from simulai.regression import SLFNN, ConvexDenseNetwork
             from simulai.models import ImprovedDenseNetwork
+            from simulai.regression import SLFNN, ConvexDenseNetwork
 
             # Configuration for the fully-connected network
             config = {