lint

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/control/proc/zoe/zoedepth/models/base_models/midas_repo/midas/backbones/next_vit.py

@@ -1,7 +1,7 @@
 import timm
 import torch.nn as nn
 from .utils import activations, forward_default, get_activation
-from ..external.next_vit.classification.nextvit import * # noqa
+from ..external.next_vit.classification.nextvit import *
 
 
 def forward_next_vit(pretrained, x):

modules/sdnq/layers/conv/conv_fp16.py

@@ -2,14 +2,14 @@
 
 import torch
 
-from ...common import compile_func, fp_mm_func # noqa: TID252
-from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias # noqa: TID252
-from ...quant_utils import rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_float import unpack_float # noqa: TID252
+from ...common import compile_func, fp_mm_func
+from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias
+from ...quant_utils import rotate_hadamard, get_hadamard
+from ...packed_float import unpack_float
 
 from .forward import get_conv_args, process_conv_input
-from ..linear.linear_fp8_tensorwise import quantize_fp_mm_input_tensorwise # noqa: TID252
-from ..linear.forward import check_mats # noqa: TID252
+from ..linear.linear_fp8_tensorwise import quantize_fp_mm_input_tensorwise
+from ..linear.forward import check_mats
 
 
 def conv_fp16_matmul(

modules/sdnq/layers/conv/conv_fp8.py

@@ -2,13 +2,13 @@
 
 import torch
 
-from ...common import compile_func # noqa: TID252
-from ...quant_utils import rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_float import unpack_float # noqa: TID252
+from ...common import compile_func
+from ...quant_utils import rotate_hadamard, get_hadamard
+from ...packed_float import unpack_float
 
 from .forward import get_conv_args, process_conv_input
-from ..linear.linear_fp8 import quantize_fp_mm_input # noqa: TID252
-from ..linear.forward import check_mats # noqa: TID252
+from ..linear.linear_fp8 import quantize_fp_mm_input
+from ..linear.forward import check_mats
 
 
 def conv_fp8_matmul(

modules/sdnq/layers/conv/conv_fp8_tensorwise.py

@@ -2,14 +2,14 @@
 
 import torch
 
-from ...common import compile_func # noqa: TID252
-from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias # noqa: TID252
-from ...quant_utils import rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_float import unpack_float # noqa: TID252
+from ...common import compile_func
+from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias
+from ...quant_utils import rotate_hadamard, get_hadamard
+from ...packed_float import unpack_float
 
 from .forward import get_conv_args, process_conv_input
-from ..linear.linear_fp8_tensorwise import quantize_fp_mm_input_tensorwise # noqa: TID252
-from ..linear.forward import check_mats # noqa: TID252
+from ..linear.linear_fp8_tensorwise import quantize_fp_mm_input_tensorwise
+from ..linear.forward import check_mats
 
 
 def conv_fp8_matmul_tensorwise(

modules/sdnq/layers/conv/conv_int8.py

@@ -2,14 +2,14 @@
 
 import torch
 
-from ...common import compile_func, int_mm_func # noqa: TID252
-from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias # noqa: TID252
-from ...quant_utils import rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_int import unpack_int # noqa: TID252
+from ...common import compile_func, int_mm_func
+from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias
+from ...quant_utils import rotate_hadamard, get_hadamard
+from ...packed_int import unpack_int
 
 from .forward import get_conv_args, process_conv_input
-from ..linear.linear_int8 import quantize_int_mm_input # noqa: TID252
-from ..linear.forward import check_mats # noqa: TID252
+from ..linear.linear_int8 import quantize_int_mm_input
+from ..linear.forward import check_mats
 
 
 def conv_int8_matmul(

modules/sdnq/layers/embedding/forward.py

@@ -2,11 +2,11 @@
 
 import torch
 
-from ...common import dtype_dict, compile_func # noqa: TID252
-from ...dequantizer import dequantize_symmetric, dequantize_asymmetric # noqa: TID252
-from ...quant_utils import get_hadamard # noqa: TID252
-from ...packed_int import unpack_int # noqa: TID252
-from ...packed_float import unpack_float # noqa: TID252
+from ...common import dtype_dict, compile_func
+from ...dequantizer import dequantize_symmetric, dequantize_asymmetric
+from ...quant_utils import get_hadamard
+from ...packed_int import unpack_int
+from ...packed_float import unpack_float
 
 
 def quantized_embedding(

modules/sdnq/layers/linear/forward.py

@@ -2,7 +2,7 @@
 
 import torch
 
-from ...common import use_contiguous_mm # noqa: TID252
+from ...common import use_contiguous_mm
 
 
 def check_mats(input: torch.Tensor, weight: torch.Tensor, allow_contiguous_mm: bool = True) -> tuple[torch.Tensor, torch.Tensor]:

modules/sdnq/layers/linear/linear_fp16.py

@@ -2,10 +2,10 @@
 
 import torch
 
-from ...common import compile_func, fp_mm_func # noqa: TID252
-from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias # noqa: TID252
-from ...quant_utils import rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_float import unpack_float # noqa: TID252
+from ...common import compile_func, fp_mm_func
+from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias
+from ...quant_utils import rotate_hadamard, get_hadamard
+from ...packed_float import unpack_float
 
 from .forward import check_mats
 from .linear_fp8_tensorwise import quantize_fp_mm_input_tensorwise

modules/sdnq/layers/linear/linear_fp8.py

@@ -2,9 +2,9 @@
 
 import torch
 
-from ...common import compile_func # noqa: TID252
-from ...quant_utils import quantize_fp_mm, rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_float import unpack_float # noqa: TID252
+from ...common import compile_func
+from ...quant_utils import quantize_fp_mm, rotate_hadamard, get_hadamard
+from ...packed_float import unpack_float
 
 from .forward import check_mats
 

modules/sdnq/layers/linear/linear_fp8_tensorwise.py

@@ -2,10 +2,10 @@
 
 import torch
 
-from ...common import compile_func # noqa: TID252
-from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias # noqa: TID252
-from ...quant_utils import quantize_fp_mm, rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_float import unpack_float # noqa: TID252
+from ...common import compile_func
+from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias
+from ...quant_utils import quantize_fp_mm, rotate_hadamard, get_hadamard
+from ...packed_float import unpack_float
 
 from .forward import check_mats
 

modules/sdnq/layers/linear/linear_int8.py

@@ -2,10 +2,10 @@
 
 import torch
 
-from ...common import compile_func, int_mm_func # noqa: TID252
-from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias # noqa: TID252
-from ...quant_utils import quantize_int_mm, rotate_hadamard, get_hadamard # noqa: TID252
-from ...packed_int import unpack_int # noqa: TID252
+from ...common import compile_func, int_mm_func
+from ...dequantizer import dequantize_symmetric, dequantize_symmetric_with_bias
+from ...quant_utils import quantize_int_mm, rotate_hadamard, get_hadamard
+from ...packed_int import unpack_int
 
 from .forward import check_mats
 

modules/sdnq/packed_int/__init__.py

@@ -1,6 +1,6 @@
 import torch
 
-from ..common import dtype_dict # noqa: TID252
+from ..common import dtype_dict
 
 from .pack import (
     pack_uint15,

pyproject.toml

@@ -68,6 +68,7 @@ ignore = [
     "RUF022", # All is not sorted
     "RUF046", # Value being cast to `int` is already an integer
     "RUF051", # Prefer pop over del
+    "TID252", # Relative imports in vendored code
 ]
 fixable = ["ALL"]
 unfixable = []