A util to list files in a folder

realfake/utils.py

@@ -120,3 +120,7 @@ def find_latest_checkpoint(dirname: Path) -> Path:
         return None
     latest = max(checkpoints, key=lambda path: path.stat().st_mtime)
     return latest
+
+
+def list_files(dirname: Path, exts: list[str]) -> list:
+    return [fn for fn in Path(dirname).iterdir() for ext in exts if fn.match(f"*.{ext}")]