update script

update-cpu-html.py

@@ -62,6 +62,7 @@ def main():
     wheels = get_all_files("macos", suffix="*.whl")
     wheels += get_all_files("ubuntu-cpu", suffix="*.whl")
     wheels += get_all_files("windows-cpu", suffix="*.whl")
+    wheels += get_all_files("cpu", suffix="*.whl")
     urls = generate_url(wheels)
 
     wheels = []

util.py

@@ -1,7 +1,8 @@
 #!/usr/bin/env python3
 
-from typing import List
+import os
 from pathlib import Path
+from typing import List
 
 BASE_URL = "https://huggingface.co/csukuangfj/kaldifeat/resolve/main/"
 
@@ -15,4 +16,19 @@ def generate_url(files: List[str]) -> List[str]:
 
 
 def get_all_files(d: str, suffix: str) -> List[str]:
-    return sorted(Path(d).glob(suffix), reverse=True)
+    if "*" in suffix:
+        suffix = suffix.split(".")[-1]
+
+    d_list = d
+
+    if isinstance(d_list, str):
+        d_list = [d_list]
+
+    ss = []
+    for d in d_list:
+        for root, _, files in os.walk(d):
+            for f in files:
+                if f.endswith(suffix):
+                    ss.append(os.path.join(root, f))
+
+    return sorted(ss, reverse=True)