From d5845f5f7bb9a840444045d6e6c71c5b89e3203c Mon Sep 17 00:00:00 2001 From: Mark Leone Date: Fri, 17 Jan 2025 08:41:10 -0800 Subject: [PATCH] updated sampling script to get rid of jolt --- stack/main/scripts/control_inputs_sampling.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/stack/main/scripts/control_inputs_sampling.py b/stack/main/scripts/control_inputs_sampling.py index 39d85fa..f209a35 100644 --- a/stack/main/scripts/control_inputs_sampling.py +++ b/stack/main/scripts/control_inputs_sampling.py @@ -34,9 +34,6 @@ def set_adiabatic_control_offset(n_samples): return const_input -# TODO: write this function -def adiabatic_jolt_sampling(control_variables, seed): - pass # TODO: maybe use check_settled instead of this simplification where I just wait a certain number of seconds # for sampling n_perturbations perturbations about a single constant equilibrium point for automatic adiabatic data collection @@ -359,8 +356,6 @@ def main(data_type, sampling_type, seed=None): control_inputs_df = adiabatic_manual_sampling(control_variables) elif sampling_type == 'adiabatic_step': control_inputs_df = adiabatic_step_sampling(control_variables, seed) - elif sampling_type == 'adiabatic_jolt': - control_inputs_df = adiabatic_jolt_sampling(control_variables, seed) else: raise ValueError(f"Invalid sampling_type: {sampling_type}") @@ -370,6 +365,6 @@ def main(data_type, sampling_type, seed=None): if __name__ == '__main__': data_type = 'dynamic' # 'steady_state' or 'dynamic' - sampling_type = 'adiabatic_step' # 'circle', 'beta', 'targeted', 'uniform', 'sinusoidal', 'adiabatic_manual', 'adiabatic_step', or 'adiabatic_jolt' + sampling_type = 'adiabatic_step' # 'circle', 'beta', 'targeted', 'uniform', 'sinusoidal', 'adiabatic_manual', or 'adiabatic_step' seed = None # choose integer seed number main(data_type, sampling_type, seed)