Ejemplo n.º 1
0
        write_done(block, 'solve_s{}'.format(solve_number), db_name, db_host)

        print("Release block")
        client.release_block(block, 0)

    return 0


if __name__ == "__main__":
    predict_config = sys.argv[1]
    worker_config = sys.argv[2]
    data_config = sys.argv[3]
    graph_config = sys.argv[4]
    solve_config = sys.argv[5]

    predict_config_dict = read_predict_config(predict_config)
    worker_config_dict = read_worker_config(worker_config)
    data_config_dict = read_data_config(data_config)
    graph_config_dict = read_graph_config(graph_config)
    solve_config_dict = read_solve_config(solve_config)

    full_config = predict_config_dict
    full_config.update(worker_config_dict)
    full_config.update(data_config_dict)
    full_config.update(graph_config_dict)
    full_config.update(solve_config_dict)

    start_time = time.time()
    solve_in_block(**full_config)
    print("Solving took {} seconds".format(time.time() - start_time))
Ejemplo n.º 2
0
import os
from subprocess import check_call
from funlib.run import run, run_singularity
import logging
import scipy.interpolate.interpnd # Workaround for cython __reduce_cython__ error
from micron import read_predict_config, read_worker_config, read_data_config, read_graph_config, read_solve_config

predict_config = read_predict_config("predict_config.ini")
worker_config = read_worker_config("worker_config.ini")
data_config = read_data_config("data_config.ini")
graph_config = read_graph_config("graph_config.ini")
solve_config = read_solve_config("solve_config.ini")

base_cmd = "python {} {} {} {} {} {}".format(solve_config["daisy_solve"],
                                             os.path.abspath("predict_config.ini"),
                                             os.path.abspath("worker_config.ini"),
                                             os.path.abspath("data_config.ini"),
                                             os.path.abspath("graph_config.ini"),
                                             os.path.abspath("solve_config.ini"))

check_call(base_cmd, shell=True)
Ejemplo n.º 3
0
from funlib.run import run, run_singularity
import logging
import scipy.interpolate.interpnd  # Workaround for cython __reduce_cython__ error
import sys
from micron import read_predict_config, read_worker_config, read_data_config, read_graph_config, read_solve_config

logger = logging.getLogger(__name__)
out_hdlr = logging.StreamHandler(sys.stdout)
out_hdlr.setFormatter(logging.Formatter('%(asctime)s %(message)s'))
out_hdlr.setLevel(logging.INFO)
logger.addHandler(out_hdlr)
logger.setLevel(logging.INFO)

directory = os.path.dirname(os.path.abspath(__file__))
predict_config_path = os.path.join(directory, "predict_config.ini")
worker_config_path = os.path.join(directory, "worker_config.ini")
data_config_path = os.path.join(directory, "data_config.ini")
graph_config_path = os.path.join(directory, "graph_config.ini")
solve_config_path = os.path.join(directory, "solve_config.ini")

solve_config = read_solve_config(solve_config_path)

base_cmd = "python {} {} {} {} {} {}".format(
    solve_config["daisy_solve"], predict_config_path, worker_config_path,
    data_config_path, graph_config_path, solve_config_path)

check_call(base_cmd,
           shell=True,
           stdout=open(os.path.join(directory, "solve.out"), "w"),
           stderr=open(os.path.join(directory, "solve.err"), "w"))