From be088a31f09905ec099fa17578d79dbfd35f7576 Mon Sep 17 00:00:00 2001
From: rlskoeser <rebecca.s.koeser@princeton.edu>
Date: Thu, 11 Jan 2024 15:03:03 -0500
Subject: [PATCH] Configure hawk/dove multi batch run to be installed as a
 script

---
 pyproject.toml                            | 4 ++++
 simulatingrisk/hawkdovemulti/batch_run.py | 6 +++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index aafbc23..afe5e81 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,6 +18,10 @@ dependencies = [
 ]
 dynamic = ["version", "readme"]
 
+[project.scripts]
+simrisk-hawkdovemulti-batchrun = "simulatingrisk.hawkdovemulti.batch_run:main"
+
+
 [tool.setuptools.dynamic]
 version = {attr = "simulatingrisk.__version__"}
 readme = {file = ["README.md"]}
diff --git a/simulatingrisk/hawkdovemulti/batch_run.py b/simulatingrisk/hawkdovemulti/batch_run.py
index d912aac..c08b8b2 100755
--- a/simulatingrisk/hawkdovemulti/batch_run.py
+++ b/simulatingrisk/hawkdovemulti/batch_run.py
@@ -128,7 +128,7 @@ def batch_run(params, iterations, number_processes, max_steps, progressbar):
                     pbar.update()
 
 
-if __name__ == "__main__":
+def main():
     parser = argparse.ArgumentParser(
         prog="hawk/dove batch_run",
         description="Batch run for hawk/dove multi risk attitude simulation",
@@ -163,3 +163,7 @@ def batch_run(params, iterations, number_processes, max_steps, progressbar):
     )
     args = parser.parse_args()
     batch_run(params, args.iterations, args.processes, args.max_steps, args.progress)
+
+
+if __name__ == "__main__":
+    main()