Fixes for AMPLIFY QA scripts (#825)

pstjohn · trvachov · web-flow · commit 4ab5a214e269 · 2025-04-13T20:45:05.000-04:00
Some minor bugfixes in testing QA commands

Signed-off-by: Peter St. John &lt;pstjohn@nvidia.com&gt;
Co-authored-by: Timur Rvachov &lt;120140748+trvachov@users.noreply.github.com&gt;
diff --git a/sub-packages/bionemo-amplify/pyproject.toml b/sub-packages/bionemo-amplify/pyproject.toml
@@ -29,8 +29,8 @@ te = [
 ]
 
 [project.scripts]
-train_amplify = "bionemo.amplify.train_amplify:main"
-infer_amplify = "bionemo.amplify.infer_amplify:main"
+train_amplify = "bionemo.amplify.train_amplify:app"
+infer_amplify = "bionemo.amplify.infer_amplify:app"
 
 [tool.setuptools.packages.find]
 where = ["src"]

Original file line number	Diff line number	Diff line change
`@@ -29,8 +29,8 @@ te = [`
`29`	`29`	`]`
`30`	`30`
`31`	`31`	`[project.scripts]`
`32`		`-train_amplify = "bionemo.amplify.train_amplify:main"`
`33`		`-infer_amplify = "bionemo.amplify.infer_amplify:main"`
	`32`	`+train_amplify = "bionemo.amplify.train_amplify:app"`
	`33`	`+infer_amplify = "bionemo.amplify.infer_amplify:app"`
`34`	`34`
`35`	`35`	`[tool.setuptools.packages.find]`
`36`	`36`	`where = ["src"]`