Specify optional dependencies for shortfin_apps (#513)

nod-ai · Nov 14, 2024 · 7cc1db3 · 7cc1db3
1 parent 4fcf77b
commit 7cc1db3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/shark-ai/pyproject.toml b/shark-ai/pyproject.toml
@@ -30,6 +30,9 @@ Repository = "https://github.com/nod-ai/SHARK-Platform"
 onnx = [
   "iree-base-compiler[onnx]",
 ]
+apps = [
+  "shortfin[apps]",
+]
 
 [tool.setuptools]
 packages = []

diff --git a/shortfin/pyproject.toml b/shortfin/pyproject.toml
@@ -34,6 +34,13 @@ dynamic = ["version"]
 Repository = "https://github.com/nod-ai/SHARK-Platform"
 Documentation = "https://shortfin.readthedocs.io/en/latest/"
 
+[project.optional-dependencies]
+apps = [
+  "transformers",
+  "dataclasses-json",
+  "pillow",
+]
+
 [tool.pytest.ini_options]
 addopts = [
     "-ra",