| | 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) |
| |
|
