ruff --select=I --fix haystack/modeling/infer.py

haystack/modeling/infer.py

@@ -1,21 +1,20 @@
-from typing import List, Optional, Dict, Union, Set, Any
-
-import os
+import contextlib
 import logging
-from tqdm import tqdm
+import os
+from typing import Any, Dict, List, Optional, Set, Union
+
 import torch
-from torch.utils.data.sampler import SequentialSampler
 from torch.utils.data import Dataset
+from torch.utils.data.sampler import SequentialSampler
+from tqdm import tqdm
 
 from haystack.modeling.data_handler.dataloader import NamedDataLoader
-from haystack.modeling.data_handler.processor import Processor, InferenceProcessor
-from haystack.modeling.data_handler.samples import SampleBasket
-from haystack.modeling.utils import initialize_device_settings, set_all_seeds
 from haystack.modeling.data_handler.inputs import QAInput
+from haystack.modeling.data_handler.processor import InferenceProcessor, Processor
+from haystack.modeling.data_handler.samples import SampleBasket
 from haystack.modeling.model.adaptive_model import AdaptiveModel, BaseAdaptiveModel
 from haystack.modeling.model.predictions import QAPred
-import contextlib
-
+from haystack.modeling.utils import initialize_device_settings, set_all_seeds
 
 logger = logging.getLogger(__name__)
 

pyproject.toml

@@ -336,11 +336,11 @@ disable = [
 ]
 [tool.pylint.'DESIGN']
 max-args = 38  # Default is 5
-max-attributes = 27  # Default is 7
+max-attributes = 28  # Default is 7
 max-branches = 34  # Default is 12
 max-locals = 45  # Default is 15
 max-module-lines = 2468  # Default is 1000
-max-nested-blocks = 7  # Default is 5
+max-nested-blocks = 9  # Default is 5
 max-statements = 206  # Default is 50
 [tool.pylint.'SIMILARITIES']
 min-similarity-lines=6