Configure hawk/dove multi batch run to be installed as a script

Princeton-CDH · Jan 11, 2024 · be088a3 · be088a3
1 parent 0a77e59
commit be088a3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git 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
@@ -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()