removed AutoGPTQ dep in linting

foundation-model-stack · May 28, 2024 · 4719982 · 4719982
1 parent 9888564
commit 4719982
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/plugins/accelerated-peft/src/fms_acceleration_peft/framework_plugin_autogptq.py b/plugins/accelerated-peft/src/fms_acceleration_peft/framework_plugin_autogptq.py
@@ -51,8 +51,8 @@ def model_loader(self, model_name: str, **kwargs):
 
         # guarded imports
         # Third Party
-        from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig #pylint: disable=import-outside-toplevel
-        from auto_gptq.nn_modules.qlinear.qlinear_tritonv2 import QuantLinear #pylint: disable=import-outside-toplevel
+        from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig #pylint: disable=import-outside-toplevel,import-error
+        from auto_gptq.nn_modules.qlinear.qlinear_tritonv2 import QuantLinear #pylint: disable=import-outside-toplevel,import-error
 
         # Local
         from .autogptq_utils import patch_forward_to_view_attributes_before_call #pylint: disable=import-outside-toplevel
@@ -202,8 +202,8 @@ def augmentation(
     ):
         # guarded imports
         # Third Party
-        from auto_gptq.nn_modules.qlinear.qlinear_tritonv2 import QuantLinear #pylint: disable=import-outside-toplevel
-        from auto_gptq.utils.peft_utils import GPTQLoraModel, get_gptq_peft_model #pylint: disable=import-outside-toplevel
+        from auto_gptq.nn_modules.qlinear.qlinear_tritonv2 import QuantLinear #pylint: disable=import-outside-toplevel,import-error
+        from auto_gptq.utils.peft_utils import GPTQLoraModel, get_gptq_peft_model #pylint: disable=import-outside-toplevel,import-error
 
         # Local
         from .autogptq_utils import create_new_module_peft, replace_module_peft #pylint: disable=import-outside-toplevel

diff --git a/plugins/accelerated-peft/tox.ini b/plugins/accelerated-peft/tox.ini
@@ -20,7 +20,7 @@ deps =
     pytest>=7
     pylint>=2.16.2,<=3.1.0
 commands =
-    pip install -e {toxinidir}
+    pip install --no-deps {toxinidir}
     pylint src tests
 allowlist_externals = pylint