diff --git a/fleetrec/core/utils/envs.py b/fleetrec/core/utils/envs.py index 53d68ce4c43be2cc468a9118d0ffdcf498a761a2..79383172ca6dea8bccddba0e27a9d792d3ae7876 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 4fe2841ac587c0dab271f4d0c06a002a64e5cf0d..2dc8e1dc90abb655c470fd879dbe1f62e1b55d6a 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")