Fix itertools import

SeldonIO · Oct 31, 2019 · f14ccef · f14ccef
1 parent 60108bd
commit f14ccef
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -1,5 +1,5 @@
 import os
-from itertools.chain import from_iterable
+from itertools import chain
 from setuptools import find_packages, setup
 
 version = {}
@@ -9,7 +9,8 @@
 
 # Extra dependencies, with special 'all' key
 extras = {"tensorflow": ["tensorflow"], "gcs": ["google-cloud-storage >= 1.16.0"]}
-extras["all"] = list(set(from_iterable(extras.values())))
+all_extra_deps = chain.from_iterable(extras.values())
+extras["all"] = list(set(all_extra_deps))
 
 setup(
     name="seldon-core",