| import os |
| import sys |
| import time |
|
|
| from code_generation import generate_and_debug, prepare_working_folder |
|
|
| def repeated_sample(cfg): |
| num_repeated_samples = cfg.method.num_repeated_samples |
| num_trials = cfg.method.num_debugging_trials_per_sample |
| pde_name = cfg.pde.name |
| working_folder = cfg.working_folder |
| model_name = cfg.model.name |
|
|
| if not os.path.exists(working_folder): |
| os.makedirs(working_folder) |
|
|
| if cfg.redirect_stdout: |
| sys.stdout = open(os.path.join(working_folder, 'stdout.txt'), 'w') |
|
|
| print(f'Model name: {cfg.model.name}') |
| print(f'Working folder: {working_folder}') |
|
|
| prepare_working_folder( |
| cfg, |
| working_folder=working_folder, |
| pde_name=pde_name, |
| use_sample_solver_init=False |
| ) |
|
|
|
|
| for sample_idx in range(num_repeated_samples): |
| try: |
| generate_and_debug( |
| cfg, |
| round_idx=sample_idx, |
| num_trials=num_trials, |
| pde_name=pde_name, |
| working_folder=working_folder, |
| seed_implementations=None, |
| model_name=model_name |
| ) |
| except Exception as e: |
| print(f'Error in sample {sample_idx}: {e}. Move on to the next sample.') |
| |
| time.sleep(2) |
|
|
