update msmarco eval

Author: 545999961

FlagEmbedding/evaluation/msmarco/__main__.py

@@ -5,19 +5,22 @@
     MSMARCOEvalRunner
 )
 
+def main():
+    parser = HfArgumentParser((
+        MSMARCOEvalArgs,
+        MSMARCOEvalModelArgs
+    ))
 
-parser = HfArgumentParser((
-    MSMARCOEvalArgs,
-    MSMARCOEvalModelArgs
-))
+    eval_args, model_args = parser.parse_args_into_dataclasses()
+    eval_args: MSMARCOEvalArgs
+    model_args: MSMARCOEvalModelArgs
 
-eval_args, model_args = parser.parse_args_into_dataclasses()
-eval_args: MSMARCOEvalArgs
-model_args: MSMARCOEvalModelArgs
+    runner = MSMARCOEvalRunner(
+        eval_args=eval_args,
+        model_args=model_args
+    )
 
-runner = MSMARCOEvalRunner(
-    eval_args=eval_args,
-    model_args=model_args
-)
+    runner.run()
 
-runner.run()
+if __name__ == "__main__":
+    main()