From ebd1d64fb3b95a008ce789f76f9f120a4423acfa Mon Sep 17 00:00:00 2001 From: tangwei Date: Wed, 15 Apr 2020 19:29:35 +0800 Subject: [PATCH] update setup.py --- fleetrec/core/utils/envs.py | 2 +- fleetrec/run.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/fleetrec/core/utils/envs.py b/fleetrec/core/utils/envs.py index 53d68ce4..79383172 100644 --- a/fleetrec/core/utils/envs.py +++ b/fleetrec/core/utils/envs.py @@ -22,7 +22,7 @@ def set_runtime_envions(envs): assert isinstance(envs, dict) for k, v in envs.items(): - os.environ[k] = v + os.environ[k] = str(v) def get_runtime_envion(key): diff --git a/fleetrec/run.py b/fleetrec/run.py index 4fe2841a..2dc8e1dc 100644 --- a/fleetrec/run.py +++ b/fleetrec/run.py @@ -20,13 +20,14 @@ def single_engine(single_envs, model_yaml): def local_cluster_engine(cluster_envs, model_yaml): print(envs.pretty_print_envs(cluster_envs, ("Local Cluster Envs", "Value"))) - + envs.set_runtime_envions(cluster_envs) launch = local_engine.Launch(cluster_envs, model_yaml) launch.run() def local_mpi_engine(cluster_envs, model_yaml): print(envs.pretty_print_envs(cluster_envs, ("Local MPI Cluster Envs", "Value"))) + envs.set_runtime_envions(cluster_envs) print("coming soon") -- GitLab