use .is_binary

nanograv · Oct 30, 2023 · 9d59661 · 9d59661
1 parent 6802825
commit 9d59661
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/pint/pintk/plk.py b/src/pint/pintk/plk.py
@@ -14,7 +14,6 @@
 
 import pint.pintk.pulsar as pulsar
 import pint.pintk.colormodes as cm
-from pint.models.pulsar_binary import PulsarBinary
 from pint.models.astrometry import Astrometry
 
 import tkinter as tk
@@ -536,9 +535,7 @@ def setChoice(self, xid="mjd", yid="pre-fit"):
             ):
                 self.xbuttons[ii].configure(state="disabled")
                 self.ybuttons[ii].configure(state="disabled")
-            elif choice == "orbital phase" and not any(
-                isinstance(x, PulsarBinary) for x in model.components
-            ):
+            elif choice == "orbital phase" and not model.is_binary:
                 self.xbuttons[ii].configure(state="disabled")
                 self.ybuttons[ii].configure(state="disabled")
             if choice == "frequency" and (