diff --git a/shark-ai/pyproject.toml b/shark-ai/pyproject.toml index 5a7493ec7..f78a1641f 100644 --- 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 index 15bd68732..67483fb05 100644 --- 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",