Collect .conda packages in glob for upload (#113)

rapidsai · Aug 30, 2024 · e94044f · e94044f
1 parent 2b3d27e
commit e94044f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tools/rapids-find-anaconda-uploads.py b/tools/rapids-find-anaconda-uploads.py
@@ -64,7 +64,10 @@ def file_filter_fn(file_path):
 
 if __name__ == "__main__":
     directory_to_search = sys.argv[1]
-    tar_files = glob.glob(f"{directory_to_search}/**/*.tar.bz2", recursive=True)
+    conda_pkgs = (
+        glob.glob(f"{directory_to_search}/**/*.conda", recursive=True) +
+        glob.glob(f"{directory_to_search}/**/*.tar.bz2", recursive=True)
+    )
 
-    filtered_list = list(filter(file_filter_fn, tar_files))
+    filtered_list = list(filter(file_filter_fn, conda_pkgs))
     print("\n".join(filtered_list))