From d7c62661a314c631b3bbf6405143934c8c3e8b5f Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <morgan@huggingface.co>
Date: Tue, 17 Dec 2019 11:23:39 +0100
Subject: [PATCH] Provide serving dependencies for tensorflow and pytorch
 (serving-tf, serving-torch)

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 0b7e512955..b3b6e2e063 100644
--- 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()]