Mixer tunning code

class MixerTuner(object):
    def __init__(self, qm, sa, element, lo_freq, if_freq, ref_power):
        self.qm = qm
        self.element = element
        self.sa = sa 
        self.lo_freq = lo_freq
        self.if_freq = if_freq
        self.ref_power = ref_power
        self.job = self.init_qm()  
    def init_qm(self):
        with program() as cw:
            with infinite_loop_():
                play("CW", element)
        self.job = self.qm.execute(cw)
        return self.job
    
    def close(self):
        self.job.halt()
    
    def no_dc_offset_correction(self):
        self.qm.set_output_dc_offset_by_element(self.element, "I", float(0)) 
        self.qm.set_output_dc_offset_by_element(self.element, "Q", float(0))
        self.qm.set_mixer_correction("mixer_qubit", qubit_IF, qubit_LO, IQ_imbalance(0, 0))
        freqs, amps = self.sa.sweep(center=self.lo_freq, span= 250e6, ref_power = self.ref_power)
        plt.figure()
        plt.plot(freqs, amps)
        
    def offset_lo_leakage(self, fatol=1):  
        # fatol = 1dB
        print('Offset Lo leakage')
        freq_i, amps_i = sa.sweep(center=self.lo_req, span= 250e6, ref_power = self.ref_power)
        plt.plot(freqs_i, amps_i)
        
        def objective_func(offset_I, offset_Q):
            self.qm.set_output_dc_offset_by_element(self.element, "I", float(offset_I)) 
            self.qm.set_output_dc_offset_by_element(self.element, "Q", float(offset_Q))
        
            freqs, amps = self.sa.sweep(center=self.lo_req, span= 250e6, ref_power = self.ref_power)
            
            # around postion of lo, required_sideband, removed_sideband
            # np.searchsorted(list, threshold, side='right') most fast way 
            # index_required_sideband = np.argmax(freq>= (self.lo_freq+self.if_freq))
            # index_removed_sideband = np.argmax(freq>= (self.lo_freq-self.if_freq))
            
            threshold = np.mean(amps)
            index_lo = np.argmax(freq>= self.lo_freq)
            lo_peak_amps = amps[index_lo]
            contrast = lo_peak_amps - threshold
            return np.abs(contrast) 
        
        result = sc.optimize.minimize(objective_func, [0, 0], method='Nelder-Mead', fatol= fatol)
        
        if result.success:
            fitted_params = result.x
            print("The I Q offsets are ",fitted_params)

            self.qm.set_output_dc_offset_by_element(self.element, "I", float(fitted_params[0])) 
            self.qm.set_output_dc_offset_by_element(self.element, "Q", float(fitted_params[1]))
            freqs_f, amps_f = self.sa.sweep(center=self.lo_req, span= 250e6, ref_power = self.ref_power)
            plt.plot(freqs_f, amps_f)
        else:
            raise ValueError(result.message)
        
    def offset_iq_imbalance(self, fatol=1):  
        # fatol = 1dB
        print('Offset IQ imbalance')
        freq_i, amps_i = sa.sweep(center=self.lo_req, span= 250e6, ref_power = self.ref_power)
        plt.plot(freqs_i, amps_i)
        
        
        def objective_func(gain, phase):
            self.qm.set_mixer_correction("mixer_"+element, int(self.if_freq), int(self.lo_freq), IQ_imbalance(gain, phase))
        
            freqs, amps = self.sa.sweep(center=self.lo_req, span= 250e6, ref_power = self.ref_power)
            
            # around postion of lo, required_sideband, removed_sideband
            # np.searchsorted(list, threshold, side='right') most fast way 
            # index_required_sideband = np.argmax(freq>= (self.lo_freq+self.if_freq))
            index_removed_sideband = np.argmax(freq>= (self.lo_freq-self.if_freq))
            
            threshold = np.mean(amps)
            removed_sideband_peak_amps = amps[index_removed_sideband]
            contrast = removed_sideband_peak_amps - threshold
            return np.abs(contrast) 
        
        result = sc.optimize.minimize(objective_func, [0, 0], method='Nelder-Mead', fatol= fatol)
        
        if result.success:
            fitted_params = result.x
            print("The mixer correction is ",fitted_params)
            fitted_params
            self.qm.set_mixer_correction("mixer_"+element, int(self.if_freq), int(self.lo_freq), IQ_imbalance(fitted_params[0], fitted_params[0]))
            
            freqs_f, amps_f = self.sa.sweep(center=self.lo_req, span= 200e6, ref_power = self.ref_power)
            plt.plot(freqs_f, amps_f)

        else:
            raise ValueError(result.message)