fix bug

fleetrec/core/utils/envs.py

@@ -22,6 +22,10 @@ def set_runtime_envions(envs):
     assert isinstance(envs, dict)
 
     def fatten_env_namespace(namespace_nests, local_envs):
+        if not isinstance(local_envs, dict):
+            global_k = ".".join(namespace_nests)
+            os.environ[global_k] = str(local_envs)
+        else:
             for k, v in local_envs.items():
                 if isinstance(v, dict):
                     nests = copy.deepcopy(namespace_nests)

fleetrec/run.py

@@ -26,7 +26,7 @@ def set_runtime_envs(cluster_envs, engine_yaml):
 
     need_print = {}
     for k, v in os.environ.items():
-        if k.startswith() == "trainer.":
+        if k.startswith("trainer."):
             need_print[k] = v
 
     print(envs.pretty_print_envs(need_print, ("Runtime Envs", "Value")))