from pathlib import Path | |
import mkdocs_gen_files | |
skip_list = ["init", "python_tokenizer"] | |
src_root = Path("src/transformer_deploy") | |
for path in src_root.glob("**/*.py"): | |
if any([s in str(path) for s in skip_list]): | |
continue | |
doc_path = Path("reference", path.relative_to(src_root)).with_suffix(".md") | |
with mkdocs_gen_files.open(doc_path, "w") as f: | |
ident = ".".join(path.with_suffix("").parts) | |
print("::: " + ident, file=f) | |