def generate_dataset_osc(theta): osc = Oscillator(deg_rate=theta[0]) osc.run() return osc.ds
def generate_dataset_osc(theta): osc = Oscillator(deg_rate=theta[0], rep_strength=theta[1]) osc.run() return osc.ds