fix test

nanograv · Nov 3, 2023 · 0bb9089 · 0bb9089
1 parent 8e3ef3b
commit 0bb9089
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/pint/simulation.py b/src/pint/simulation.py
@@ -385,7 +385,11 @@ def make_fake_toas_fromMJDs(
         times = MJDs
         freq_array = np.tile(freqs, len(MJDs) // len(freqs) + 1)[: len(times)]
     else:
-        times = np.repeat(MJDs, len(freqs))
+        times = (
+            time.Time(np.repeat(MJDs, len(freqs)))
+            if isinstance(MJDs, time.Time)
+            else np.repeat(MJDs, len(freqs))
+        )
         freq_array = np.tile(freqs, len(MJDs))
 
     clk_version = get_fake_toa_clock_versions(