Split model list on modality (#18328)

* 📝 split up model list

* Adapt script to reorg

* apply niels feedback

Co-authored-by: Sylvain Gugger <Sylvain.gugger@gmail.com>
This commit is contained in:
Steven Liu
2022-08-01 09:10:20 -07:00
committed by GitHub
parent 01db72abd4
commit 151a2aaa4e
2 changed files with 307 additions and 282 deletions

View File

@@ -65,12 +65,22 @@ def check_model_doc(overwrite=False):
while api_doc[model_idx]["title"] != "Models":
model_idx += 1
old_model_doc = api_doc[model_idx]["sections"]
new_model_doc = clean_model_doc_toc(old_model_doc)
model_doc = api_doc[model_idx]["sections"]
if old_model_doc != new_model_doc:
modalities_docs = [(idx, section) for idx, section in enumerate(model_doc) if "sections" in section]
diff = False
for idx, modality_doc in modalities_docs:
old_modality_doc = modality_doc["sections"]
new_modality_doc = clean_model_doc_toc(old_modality_doc)
if old_modality_doc != new_modality_doc:
diff = True
if overwrite:
model_doc[idx]["sections"] = new_modality_doc
if diff:
if overwrite:
api_doc[model_idx]["sections"] = new_model_doc
api_doc[model_idx]["sections"] = model_doc
content[api_idx]["sections"] = api_doc
with open(PATH_TO_TOC, "w", encoding="utf-8") as f:
f.write(yaml.dump(content, allow_unicode=True))