Fix typing for None valued variables (#37004)

Fix typing for None-able variables
This commit is contained in:
cyyever
2025-03-27 22:46:32 +08:00
committed by GitHub
parent 8c5e29bad5
commit de77f5b1ec
80 changed files with 271 additions and 249 deletions

View File

@@ -19,6 +19,7 @@ import json
import os
import re
from contextlib import contextmanager
from typing import Optional
from transformers.utils.import_utils import export
@@ -284,7 +285,7 @@ def model_addition_debugger(cls):
@export(backends=("torch",))
@contextmanager
def model_addition_debugger_context(model, debug_path: str = None):
def model_addition_debugger_context(model, debug_path: Optional[str] = None):
"""
# Model addition debugger - context manager for model adders
This context manager is a power user tool intended for model adders.