[install] only add deepspeed pkg at install (microsoft#2714)

Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
stas00 · Jan 18, 2023 · 0b549ad · 0b549ad
1 parent df2495c
commit 0b549ad
Showing 1 changed file with 2 additions and 14 deletions.
diff --git a/setup.py b/setup.py
@@ -281,20 +281,8 @@ def create_dir_symlink(src, dest):
       },
       install_requires=install_requires,
       extras_require=extras_require,
-      packages=find_packages(exclude=[
-          "azure",
-          "csrc",
-          "docker",
-          "docs",
-          "examples",
-          "op_builder",
-          "release",
-          "requirements",
-          "scripts",
-          "tests",
-          "benchmarks",
-          "accelerator"
-      ]),
+      packages=find_packages(include=['deepspeed',
+                                      'deepspeed.*']),
       include_package_data=True,
       scripts=[
           'bin/deepspeed',