Become agnostic to the units of PINT phases

mhvk · May 27, 2020 · 3d7e4cd · 3d7e4cd
1 parent 719eecf
commit 3d7e4cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scintillometry/phases/core.py b/scintillometry/phases/core.py
@@ -63,7 +63,8 @@ def __call__(self, t):
         shape = getattr(toas, 'shape', ())
         # TODO: Once PINT uses the Phase class, we can return the
         # result directly.
-        return Phase(ph.int, ph.frac).reshape(shape)
+        with u.add_enabled_equivalencies([(u.dimensionless_unscaled, u.cy)]):
+            return Phase(ph.int, ph.frac).reshape(shape)
 
     def apparent_spin_freq(self, t):
         """Compute the apparent spin frequency at one or more times.