format

openpsi-project · Aug 28, 2024 · 2c07b88 · 2c07b88
1 parent 11e0e7a
commit 2c07b88
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/realhf/experiments/common/gen_exp.py b/realhf/experiments/common/gen_exp.py
@@ -89,7 +89,10 @@ def rpcs(self):
         # Customized dataclass objects will not work in that case.
         interface = ModelInterfaceAbstraction(
             "generation",
-            args={"generation_config": OmegaConf.to_container(self.gen, resolve=True), "output_file": self.output_file},
+            args={
+                "generation_config": OmegaConf.to_container(self.gen, resolve=True),
+                "output_file": self.output_file,
+            },
         )
         gen = MFCDef(
             name="gen",

diff --git a/realhf/system/master_worker.py b/realhf/system/master_worker.py
@@ -1366,9 +1366,7 @@ def _poll(self):
                     "evaluate",
                     [None for _ in self.__all_model_handlers],
                 )
-                eval_stats = _gather_stat(
-                    list(filter(lambda x: bool(x), eval_stats))
-                )
+                eval_stats = _gather_stat(list(filter(lambda x: bool(x), eval_stats)))
                 logger.info(
                     f"Evaluation results at epoch {self._epoch} step {self._epoch_step}: {eval_stats}"
                 )