From f1232b35ba00d043f316a916f8cbb4bf35714ad8 Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Mon, 29 May 2023 18:04:17 +0900
Subject: [PATCH] Update mypy dependencies

---
 .mypy.ini               | 30 ++++++++++++++++++++++++++++++
 .pre-commit-config.yaml |  3 +--
 2 files changed, 31 insertions(+), 2 deletions(-)

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',
         ]