Omega definition shouldn't be in ecc block

exoplanet-dev · May 23, 2019 · a0b9c0c · a0b9c0c
1 parent 1ca5c8b
commit a0b9c0c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/exoplanet/orbits/keplerian.py b/exoplanet/orbits/keplerian.py
@@ -108,6 +108,13 @@ def __init__(self,
         if contact_points_kwargs is None:
             contact_points_kwargs = dict()
 
+        if Omega is None:
+            self.Omega = None
+        else:
+            self.Omega = tt.as_tensor_variable(Omega)
+            self.cos_Omega = tt.cos(self.Omega)
+            self.sin_Omega = tt.sin(self.Omega)
+
         # Eccentricity
         self.contact_points_op = ContactPointsOp(**contact_points_kwargs)
         if ecc is None:
@@ -124,13 +131,6 @@ def __init__(self,
             self.cos_omega = tt.cos(self.omega)
             self.sin_omega = tt.sin(self.omega)
 
-            if Omega is None:
-                self.Omega = None
-            else:
-                self.Omega = tt.as_tensor_variable(Omega)
-                self.cos_Omega = tt.cos(self.Omega)
-                self.sin_Omega = tt.sin(self.Omega)
-
             opsw = 1 + self.sin_omega
             E0 = 2 * tt.arctan2(tt.sqrt(1-self.ecc)*self.cos_omega,
                                 tt.sqrt(1+self.ecc)*opsw)