More compatibility fixes with the example code in the paper.

hpparvi · Jul 24, 2020 · c5aaffc · c5aaffc
1 parent fe8ac97
commit c5aaffc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pytransit/__init__.py b/pytransit/__init__.py
@@ -50,14 +50,14 @@
 from .models.ma_uniform import UniformModel
 from .models.ma_chromosphere import ChromosphereModel
 from .models.general import GeneralModel
-from .models.swiftmodel import SwiftModel
+from .models.swiftmodel import SwiftModel, SwiftModel as SWIFTModel
 
 # OpenCL models
 # -------------
 from .models.qpower2_cl import QPower2ModelCL
 from .models.ma_quadratic_cl import QuadraticModelCL
 from .models.ma_uniform_cl import UniformModelCL
-from .models.swiftmodel_cl import SwiftModelCL
+from .models.swiftmodel_cl import SwiftModelCL, SwiftModelCL as SWIFTModelCL
 
 from .models.ldtkldm import LDTkLDModel, LDTkLD
 

diff --git a/pytransit/models/ldtkldm.py b/pytransit/models/ldtkldm.py
@@ -85,15 +85,15 @@ def _integrate(self, x: ndarray) -> ndarray:
 
 
 class LDTkLDModel(LDModel):
-    def __init__(self, teff: Tuple[float, float], logg: Tuple[float, float], z: Tuple[float, float], pbs: Tuple,
-                 nsamples: int = 500, frozen: bool = False, cache: Optional[Union[str, Path]] = None, lowres: bool = True):
+    def __init__(self, pbs: Tuple, teff: Tuple[float, float], logg: Tuple[float, float], z: Tuple[float, float],
+                 samples: int = 500, frozen: bool = False, cache: Optional[Union[str, Path]] = None, lowres: bool = True):
         super().__init__()
         self._sc = LDPSetCreator(teff, logg, z, pbs, cache=cache, lowres=lowres)
-        self._ps = self._sc.create_profiles(nsamples)
+        self._ps = self._sc.create_profiles(samples)
         self._i = 0
 
         self.npb = len(pbs)
-        self.nsamples = nsamples
+        self.nsamples = samples
         self.frozen = frozen
         self.z = self._ps._z
         self.mu = self._ps._mu