diff --git a/.mypy.ini b/.mypy.ini index 486dde3fe..941046ae8 100644 --- a/.mypy.ini +++ b/.mypy.ini @@ -1,3 +1,33 @@ [mypy] exclude = venv + +[mypy-alpaca_lora_4bit.*] +ignore_missing_imports = True + +[mypy-flash_attn.*] +ignore_missing_imports = True + +[mypy-huggingface_hub] +ignore_missing_imports = True + +[mypy-transformers.*] +ignore_missing_imports = True + +[mypy-peft] +ignore_missing_imports = True + +[mypy-bitsandbytes] +ignore_missing_imports = True + +[mypy-datasets] +ignore_missing_imports = True + +[mypy-fire] +ignore_missing_imports = True + +[mypy-setuptools] +ignore_missing_imports = True + +[mypy-addict] +ignore_missing_imports = True diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 46e29e7a0..c578dbc67 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -26,6 +26,5 @@ repos: - id: mypy additional_dependencies: [ - 'fire', - 'types-PyYAML' + 'types-PyYAML', ]