Remove dependency on gatspy

feder-observatory · Oct 27, 2023 · d7dfdf7 · d7dfdf7
1 parent 5d73d59
commit d7dfdf7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -22,7 +22,6 @@ install_requires =
     photutils >=1
     matplotlib
     pandas
-    gatspy
     pyyaml
     astrowidgets
     ipyfilechooser

diff --git a/stellarphot/visualization/multi_night_plots.py b/stellarphot/visualization/multi_night_plots.py
@@ -4,8 +4,7 @@
 
 from astropy.stats import mad_std
 from astropy.time import Time
-
-from gatspy.periodic import LombScargleFast
+from astropy.timeseries import LombScargle
 
 
 __all__ = ['plot_magnitudes', 'multi_night']
@@ -201,11 +200,13 @@ def multi_night(sources, unique_nights, night,
                     np.isinf(mag_err[source.id - 1]))
         bads = bad_mags | bad_errs
         good_mags = ~bads
-        model = LombScargleFast().fit(source.bjd_tdb[good_mags],
-                                      mags[source.id - 1][good_mags],
-                                      mag_err[source.id - 1][good_mags])
-        periods, power = model.periodogram_auto(nyquist_factor=100,
-                                                oversampling=3)
+        ls = LombScargle(source.bjd_tdb[good_mags],
+                         mags[source.id - 1][good_mags],
+                         mag_err[source.id - 1][good_mags])
+
+        freqs, power = ls.autopower(nyquist_factor=100,
+                                    samples_per_peak=10)
+
         max_pow = power.max()
 
         # print(source, max_pow)