set rot_interp automatically

TomHilder · Jul 23, 2023 · b668d63 · b668d63
1 parent 66051e6
commit b668d63
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/wakeflow/model_setup.py b/src/wakeflow/model_setup.py
@@ -125,7 +125,6 @@ def __init__(self, config: dict) -> None:
         self.box_warp             = bool (config["box_warp"])
         self.use_box_IC           = bool (config["use_box_IC"])
         self.tf_fac               = float(config["tf_fac"])
-        self.rot_interp           = bool (config["rot_interp"])
         self.r_cut_inner_fac      = float(config["r_cut_inner_fac"])
 
         # Choice of physics
@@ -151,6 +150,12 @@ def __init__(self, config: dict) -> None:
         #    self.a_cw = -1
         #else:
         self.a_cw = 1
+
+        # check if phi planet is not zero
+        if self.phi_planet != 0.:
+            self.rot_interp = True
+        else:
+            self.rot_interp = False
 
         # get height scale at reference radius
         self.h_ref = self.hr * self.r_ref

diff --git a/src/wakeflow/wakeflow.py b/src/wakeflow/wakeflow.py
@@ -44,7 +44,7 @@ def configure(
         r_outer:            float = 500,
         r_inner:            float = 100,
         r_planet:           float = 250,
-        phi_planet:         float = 0,
+        phi_planet:         float = 0.0,
         r_ref:              float = None,
         r_c:                float = 0,
         z_max:              float = 3,