chore(sdk): add kfp-kubernetes as kfp extras_require (kubeflow#9152)

rd-pong · Apr 26, 2023 · 953c96e · 953c96e
1 parent d2c06cc
commit 953c96e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
@@ -54,6 +54,9 @@ def read_readme() -> str:
         return f.read()
 
 
+docker = ['docker']
+kubernetes = ['kfp-kubernetes<2']
+
 setuptools.setup(
     name='kfp',
     version=find_version('kfp', '__init__.py'),
@@ -74,7 +77,8 @@ def read_readme() -> str:
     },
     install_requires=get_requirements('requirements.in'),
     extras_require={
-        'all': ['docker'],
+        'all': docker + kubernetes,
+        'kubernetes': kubernetes,
     },
     packages=setuptools.find_packages(exclude=['*test*']),
     classifiers=[