Generate: unify LogitsWarper and LogitsProcessor (#32626)

2024-08-16 11:20:41 +01:00
parent 5fd7ca7bc9
commit 70d5df6107
20 changed files with 186 additions and 623 deletions
--- a/utils/check_docstrings.py
+++ b/utils/check_docstrings.py
@@ -70,6 +70,7 @@ OBJECTS_TO_IGNORE = [
    # Deprecated
    "InputExample",
    "InputFeatures",
+    "LogitsWarper",
    # Signature is *args/**kwargs
    "TFSequenceSummary",
    "TFBertTokenizer",
--- a/utils/check_repo.py
+++ b/utils/check_repo.py
@@ -932,6 +932,7 @@ DEPRECATED_OBJECTS = [
    "LineByLineTextDataset",
    "LineByLineWithRefDataset",
    "LineByLineWithSOPTextDataset",
+    "LogitsWarper",
    "NerPipeline",
    "PretrainedBartModel",
    "PretrainedFSMTModel",