Provide serving dependencies for tensorflow and pytorch (serving-tf, serving-torch)

d7c62661 · Morgan Funtowicz · a096e2a8 · d7c62661
Commit d7c62661 authored Dec 17, 2019 by Morgan Funtowicz
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

setup.py setup.py +3 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,9 @@ from setuptools import find_packages, setup
 extras = {
-    'serving': ['uvicorn', 'fastapi']
+    'serving': ['uvicorn', 'fastapi'],
+    'serving-tf': ['uvicorn', 'fastapi', 'tensorflow'],
+    'serving-torch': ['uvicorn', 'fastapi', 'torch']
 }
 extras['all'] = [package for package in extras.values()]