Update setup.py to correctly pick up Python files

And also re-update package name in utils/build-dists.py so that it is uninstalled successfully Signed-off-by: Rushi Agrawal <[email protected]>
opensearch-project · Sep 20, 2021 · e4d0cc6 · e4d0cc6
1 parent 4e7cd3a
commit e4d0cc6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -41,10 +41,11 @@
 with open(join(base_dir, "README.md")) as f:
     long_description = f.read().strip()
 
+module_dir = package_name.replace("-", "")
 packages = [
     package
     for package in find_packages(where=".", exclude=("test_opensearchpy*",))
-    if package == package_name or package.startswith(package_name + ".")
+    if package == module_dir or package.startswith(module_dir + ".")
 ]
 
 install_requires = [

diff --git a/utils/build-dists.py b/utils/build-dists.py
@@ -161,7 +161,14 @@ def test_dist(dist):
             )
 
         # Uninstall the dist, see that we can't import things anymore
-        run(venv_python, "-m", "pip", "uninstall", "--yes", dist_name)
+        run(
+            venv_python,
+            "-m",
+            "pip",
+            "uninstall",
+            "--yes",
+            dist_name.replace("opensearchpy", "opensearch-py"),
+        )
         run(
             venv_python,
             "-c",