function import

envfluids · Jul 3, 2023 · 31316bf · 31316bf
1 parent 443019b
commit 31316bf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/py2d/Py2D_solver.py b/py2d/Py2D_solver.py
@@ -28,6 +28,7 @@
 from py2d.aposteriori_analysis import eddyTurnoverTime_2DFHIT
 from py2d.SGSModel import *
 # from py2d.uv2tau_CNN import *
+from py2d.initialize import *
 
 # Enable x64 Precision for Jax
 jax.config.update('jax_enable_x64', True)
@@ -220,7 +221,7 @@ def Py2D_solver(Re, fkx, fky, alpha, beta, NX, SGSModel_string, eddyViscosityCoe
     if readTrue:
 
         # -------------- Initialization using pertubration --------------
-        w1_hat, psi_hat, psiPrevious_hat, psiCurrent_hat = initialize_random(NX, Kx, Ky):
+        w1_hat, psi_hat, psiPrevious_hat, psiCurrent_hat = initialize_perturbation(NX, Kx, Ky):
         time = 0.0
 
     else:

diff --git a/py2d/initialize.py b/py2d/initialize.py
@@ -2,7 +2,7 @@
 import numpy as nnp
 import jax.numpy as np
 
-def initialize_random(NX, Kx, Ky):
+def initialize_perturbation(NX, Kx, Ky):
     # -------------- Initialization using pertubration --------------
     kp = 10.0
     A  = 4*np.power(kp,(-5))/(3*np.pi)
@@ -28,3 +28,4 @@ def initialize_random(NX, Kx, Ky):
 
 
 #def initialize_resume(NX, Kx, Ky):
+