Specify optional dependencies for shortfin_apps

nod-ai · Nov 14, 2024 · 7365b0b · 7365b0b
1 parent 9d45921
commit 7365b0b
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",