From 2101532f8444783c68c4d545e5fc556a14a460a0 Mon Sep 17 00:00:00 2001 From: tangwei Date: Wed, 20 May 2020 13:59:08 +0800 Subject: [PATCH] add codestyle travis --- run.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/run.py b/run.py index 4588c88f..56999935 100755 --- a/run.py +++ b/run.py @@ -52,7 +52,6 @@ def get_inters_from_yaml(file, filter): flattens = envs.flatten_environs(_envs) inters = {} - for k, v in flattens.items(): if k.startswith(filter): inters[k] = v @@ -63,14 +62,16 @@ def get_engine(args): transpiler = get_transpiler() run_extras = get_inters_from_yaml(args.model, "train.") - engine = run_extras.get("train.engine", "single") + engine = run_extras.get("train.engine", "single") engine = engine.upper() if engine not in engine_choices: raise ValueError("train.engin can not be chosen in {}".format(engine_choices)) print("engines: \n{}".format(engines)) + run_engine = engines[transpiler].get(engine, None) + return run_engine -- GitLab