feat(main): add cli and entry point to package (#16)

Using __main__ file for cli
RodrigoGonzalez · Aug 21, 2023 · 9eca289 · 9eca289
1 parent d88e2f2
commit 9eca289
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 4 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,6 +21,15 @@ classifiers = [
     "Typing :: Typed",
 ]
 
+packages = [
+    { include = "shap_app", from = "src" }
+]
+
+
+[tool.poetry.scripts]
+shap-app = 'shap_app.__main__:main'
+
+
 [tool.poetry.dependencies]
 python = ">=3.10,<3.12"
 streamlit = "^1.25.0"
@@ -58,6 +67,7 @@ seaborn = "^0.12.2"
 strictyaml = "^1.7.3"
 cmake = "^3.27.2"
 matplotlib = "^3.7.2"
+typer = "^0.9.0"
 
 
 [tool.poetry.group.dev.dependencies]

diff --git a/src/shap_app/__main__.py b/src/shap_app/__main__.py
@@ -0,0 +1,12 @@
+import os
+
+import typer
+
+
+def main():
+    typer.echo("Starting Streamlit SHAP App...")
+    os.system("streamlit run src/shap_app/app.py")
+
+
+if __name__ == "__main__":
+    typer.run(main)