提交 ebd1d64f 编写于 作者: T tangwei

update setup.py

上级 8cf79b11
...@@ -22,7 +22,7 @@ def set_runtime_envions(envs): ...@@ -22,7 +22,7 @@ def set_runtime_envions(envs):
assert isinstance(envs, dict) assert isinstance(envs, dict)
for k, v in envs.items(): for k, v in envs.items():
os.environ[k] = v os.environ[k] = str(v)
def get_runtime_envion(key): def get_runtime_envion(key):
......
...@@ -20,13 +20,14 @@ def single_engine(single_envs, model_yaml): ...@@ -20,13 +20,14 @@ def single_engine(single_envs, model_yaml):
def local_cluster_engine(cluster_envs, model_yaml): def local_cluster_engine(cluster_envs, model_yaml):
print(envs.pretty_print_envs(cluster_envs, ("Local Cluster Envs", "Value"))) 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 = local_engine.Launch(cluster_envs, model_yaml)
launch.run() launch.run()
def local_mpi_engine(cluster_envs, model_yaml): def local_mpi_engine(cluster_envs, model_yaml):
print(envs.pretty_print_envs(cluster_envs, ("Local MPI Cluster Envs", "Value"))) print(envs.pretty_print_envs(cluster_envs, ("Local MPI Cluster Envs", "Value")))
envs.set_runtime_envions(cluster_envs)
print("coming soon") print("coming soon")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册