Skip to content

Commit 217e904

Browse files
kiukchungfacebook-github-bot
authored andcommitted
Add monarch_bootstrap (monarch.bootstrap_main) as console script
Summary: side-note: looks like `monarch.bootstrap_main` used to be `monarch._monarch.hyperactor.bootstrap_main`. Adds `monarch_bootstrap` as a console script so that it is installed with the monarch wheel. This makes it simpler to pass it as the the `program` (`cmd`) to the process-allocator. Reviewed By: LucasLLC, colin2328 Differential Revision: D75325523 fbshipit-source-id: af9cb48c0dbeb5b45bdd919c23acc9f05b147ddc
1 parent 86770c5 commit 217e904

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@ def run(self):
126126
entry_points={
127127
"console_scripts": [
128128
"monarch=monarch.tools.cli:main",
129+
"monarch_bootstrap=monarch.bootstrap_main:invoke_main",
129130
],
130131
},
131132
rust_extensions=[

tools/components/hyperactor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def proc_mesh(
4545
args=[
4646
"mesh-worker",
4747
f"--port={port}",
48-
"--program=monarch_bootstrap", # TODO add monarch_boostrap as console_script and install it in Dockerfile
48+
"--program=monarch_bootstrap", # installed with monarch wheel (as console script)
4949
],
5050
num_replicas=mesh.num_hosts,
5151
resource=specs.resource(h=mesh.host_type),

0 commit comments

Comments
 (0)