ri.set_lo(1250.)

#legacy.load_heterodyne_sweep_tones(ri,(np.arange(1,129)[None,:]*7/4.+ri.lo_frequency + offsets[:,None]),
#                                    num_tone_samples=nsamp)

state = dict(field_canceling_magnet=False,
             magnetic_shield=True,
             cryostat='starcryo',
             initial_zbd_rms_voltage=rms_voltage)
state.update(**setup.state())

for heater_voltage in np.arange(0, 4, .2):
    logger.info("Measuring at %.1f volts" % heater_voltage)
    fg.set_dc_voltage(heater_voltage)
    fg.enable_output(True)
    if heater_voltage > 0:
        logger.info("Waiting 10 minutes for load to stabilize")
        time.sleep(10 * 60)
    tic = time.time()
    for lo in 830. + 190 * np.arange(0, 4):
        logger.info("Measuring at LO %.1f" % lo)
        ri.set_lo(lo)
        df = acquire.new_nc_file(suffix='scan_lo_%.1f_MHz' % lo)
        state.update(**setup.state())
        state['heater_voltage'] = heater_voltage
        swa = acquire.run_sweep(ri, (np.arange(1, 257)[None, :] * 7 / 8. +
                                     ri.lo_frequency + offsets[:, None]),
                                num_tone_samples=nsamp,
                                length_seconds=0.1,
                                state=state,