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:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user