Added time check

SSDataLab · Jul 22, 2024 · 73488c3 · 73488c3
1 parent 22d2c00
commit 73488c3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/tess_ephem/ephem.py b/src/tess_ephem/ephem.py
@@ -114,6 +114,11 @@ def predict(
             One row for each time stamp that matched a TESS observation.
         """
 
+        if not isinstance(time, Time):
+            time = Time(time)
+        if time.isscalar:
+            time = time.reshape((1,))
+
         sky = self.predict_sky(time)
         crd = SkyCoord(sky.ra, sky.dec, unit="deg")
         log.info("Started matching the ephemeris to TESS observations")