SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Deprecate models script - correctly set the model name for the doc file (#30785)

Browse Source 
* Correctly set the moel name for the doc file

* Fix up
This commit is contained in:
amyeroberts
2024-05-15 15:14:11 +01:00
committed by GitHub
parent 5ca085b882
commit 58faa7b824
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
utils/deprecate_models.py
View File

@@ -79,17 +79,13 @@ def insert_tip_to_model_doc(model_doc_path, tip_message):
def get_model_doc_path(model: str) -> Tuple[Optional[str], Optional[str]]: def get_model_doc_path(model: str) -> Tuple[Optional[str], Optional[str]]:
# Possible variants of the model name in the model doc path # Possible variants of the model name in the model doc path
model_doc_paths = [ model_names = [model, model.replace("_", "-"), model.replace("_", "")]
REPO_PATH / f"docs/source/en/model_doc/{model}.md",
# Try replacing _ with - in the model name
REPO_PATH / f"docs/source/en/model_doc/{model.replace('_', '-')}.md",
# Try replacing _ with "" in the model name
REPO_PATH / f"docs/source/en/model_doc/{model.replace('_', '')}.md",
]
for model_doc_path in model_doc_paths: model_doc_paths = [REPO_PATH / f"docs/source/en/model_doc/{model_name}.md" for model_name in model_names]
for model_doc_path, model_name in zip(model_doc_paths, model_names):
if os.path.exists(model_doc_path): if os.path.exists(model_doc_path):
return model_doc_path, model return model_doc_path, model_name
return None, None return None, None
@@ -186,6 +182,7 @@ def remove_model_references_from_file(filename, models, condition):
models (List[str]): The models to remove models (List[str]): The models to remove
condition (Callable): A function that takes the line and model and returns True if the line should be removed condition (Callable): A function that takes the line and model and returns True if the line should be removed
""" """
filename = REPO_PATH / filename
with open(filename, "r") as f: with open(filename, "r") as f:
init_file = f.read() init_file = f.read()