From 59fcb656d52aacc6d57547839290d32ce1488e5a Mon Sep 17 00:00:00 2001 From: Ryokan Ri Date: Mon, 1 Apr 2024 13:41:45 +0900 Subject: [PATCH] style: fix __main__.py --- src/jmteb/__main__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/jmteb/__main__.py b/src/jmteb/__main__.py index 7a7082d..360f5e8 100644 --- a/src/jmteb/__main__.py +++ b/src/jmteb/__main__.py @@ -44,12 +44,15 @@ def main( score_recorder.record_summary() -if __name__ == '__main__': +if __name__ == "__main__": parser = ArgumentParser(parser_mode="jsonnet") parser.add_subclass_arguments(TextEmbedder, nested_key="embedder", required=True) parser.add_argument( - "--evaluators", type=dict[str, EmbeddingEvaluator], enable_path=True, default=str(Path(__file__).parent / "configs" / "jmteb.jsonnet") + "--evaluators", + type=dict[str, EmbeddingEvaluator], + enable_path=True, + default=str(Path(__file__).parent / "configs" / "jmteb.jsonnet"), ) parser.add_argument("--config", action=ActionConfigFile, help="Path to the config file.") parser.add_argument("--save_dir", type=str, default=None, help="Directory to save the outputs")