style: upgrade the linter (#339)

* style: reformated codes

* style: reformated codes

tests/data/__init__.py

@@ -3,7 +3,6 @@ import random
 
 import datasets
 import yaml
-from huggingface_hub import snapshot_download
 
 from nunchaku.utils import fetch_or_download
 

tests/flux/test_flux_cache.py

 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/test_flux_dev.py

 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/test_flux_dev_loras.py

 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/test_flux_double_fb_cache.py

 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 
@@ -9,7 +10,7 @@ from .utils import run_test
     "use_double_fb_cache,residual_diff_threshold_multi,residual_diff_threshold_single,height,width,num_inference_steps,lora_name,lora_strength,expected_lpips",
     [
         (True, 0.09, 0.12, 1024, 1024, 30, None, 1, 0.24 if get_precision() == "int4" else 0.165),
-        (True, 0.09, 0.12, 1024, 1024, 50, None, 1, 0.24 if get_precision() == "int4" else 0.144),
+        (True, 0.09, 0.12, 1024, 1024, 50, None, 1, 0.24 if get_precision() == "int4" else 0.161),
     ],
 )
 def test_flux_dev_double_fb_cache(

tests/flux/test_flux_schnell.py

 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/test_flux_teacache.py

@@ -8,6 +8,7 @@ from diffusers.pipelines.flux.pipeline_flux import FluxPipeline
 from nunchaku import NunchakuFluxTransformer2dModel
 from nunchaku.caching.teacache import TeaCache
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import already_generate, compute_lpips, offload_pipeline
 
 

tests/flux/test_flux_tools.py

@@ -2,6 +2,7 @@ import pytest
 import torch
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/test_multiple_batch.py

@@ -2,6 +2,7 @@
 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/test_shuttle_jaguar.py

 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/test_turing.py

 import pytest
 
 from nunchaku.utils import get_precision, is_turing
+
 from .utils import run_test
 
 

tests/flux/utils.py

@@ -14,6 +14,7 @@ import nunchaku
 from nunchaku import NunchakuFluxTransformer2dModel, NunchakuT5EncoderModel
 from nunchaku.caching.diffusers_adapters import apply_cache_on_pipe
 from nunchaku.lora.flux.compose import compose_lora
+
 from ..data import get_dataset
 from ..utils import already_generate, compute_lpips, hash_str_to_int