[misc] Add sglang.bench_one_batch deprecation shim (#29082)

2026-06-23 14:57:35 -07:00
parent f74a1722e6
commit 11e7c9e0e6
2 changed files with 27 additions and 1 deletions
@@ -1000,7 +1000,7 @@ def main(server_args, bench_args):
        proc.terminate()


-if __name__ == "__main__":
+def cli_main():
    parser = argparse.ArgumentParser()
    ServerArgs.add_cli_args(parser)
    BenchArgs.add_cli_args(parser)
@@ -1018,3 +1018,7 @@ if __name__ == "__main__":
    finally:
        if server_args.tp_size != 1:
            kill_process_tree(os.getpid(), include_parent=False)
+
+
+if __name__ == "__main__":
+    cli_main()