import os

import sys

import hydra

from refine import refine

from repeated_sample import repeated_sample

from funsearch import funsearch

@hydra.main(config_path='configs', config_name='default', version_base=None)

def main(cfg):

print(f'Method: {cfg.method.name}')

print(f'Model name: {cfg.model.name}')

print(f'PDE name: {cfg.pde.name}')

print(f'Working folder: {cfg.working_folder}')

if not os.path.exists(cfg.working_folder):

os.makedirs(cfg.working_folder)

if cfg.redirect_stdout:

sys.stdout = open(os.path.join(cfg.working_folder, 'stdout.txt'), 'w')

if cfg.method.name[:6] == 'refine':

refine(cfg)

elif cfg.method.name == 'repeated_sample':

repeated_sample(cfg)

elif cfg.method.name == 'funsearch':

funsearch(cfg)

else:

raise NotImplementedError(f'Unknown method: {cfg.method.name}')

if __name__ == "__main__":

main()