From 0fd0581ed64394b8c8e56b1698d4e96b932fd372 Mon Sep 17 00:00:00 2001
From: Tristan Pinsonneault-Marotte <tristpinsm@gmail.com>
Date: Fri, 13 Sep 2024 15:36:34 -0700
Subject: [PATCH] feat(tracking): Ensure fixed tones are enabled.

---
 sodetlib/operations/tracking.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/sodetlib/operations/tracking.py b/sodetlib/operations/tracking.py
index 36fa3da5..a9cdf2f2 100644
--- a/sodetlib/operations/tracking.py
+++ b/sodetlib/operations/tracking.py
@@ -3,6 +3,8 @@
 import matplotlib.pyplot as plt
 from matplotlib.patches import Rectangle
 
+from . import uxm_setup
+
 
 def compute_tracking_quality(S, f, df, sync):
     """
@@ -554,6 +556,10 @@ def relock_tracking_setup(S, cfg, bands=None, reset_rate_khz=None, nphi0=None,
     res.find_bad_chans()
     res.save()
 
+    # if fixed tones are configured, ensure they haven't been turned off
+    for band in bands:
+        uxm_setup.turn_on_fixed_tones(S, cfg, band)
+
     is_interactive = plt.isinteractive()
     try:
         if not show_plots: