Lint

examples/experimental/openvino/bert/main.py

@@ -13,13 +13,13 @@
 from pathlib import Path
 from typing import Any, Iterable
 
+import datasets
 import evaluate
 import numpy as np
 import openvino.runtime as ov
 import torch
 import transformers
 
-import datasets
 import nncf
 from nncf.parameters import ModelType
 

tests/torch/sparsity/movement/helpers/run_recipe.py

@@ -20,6 +20,7 @@
 import torch.nn
 import torch.nn.functional as F
 import torch.utils.data
+from datasets import Dataset  # pylint: disable=no-name-in-module
 from transformers import AutoModelForAudioClassification
 from transformers import AutoModelForImageClassification
 from transformers import AutoModelForSequenceClassification
@@ -33,7 +34,6 @@
 from transformers import SwinConfig
 from transformers import Wav2Vec2Config
 
-from datasets import Dataset  # pylint: disable=no-name-in-module
 from nncf import NNCFConfig
 from nncf.experimental.torch.sparsity.movement.scheduler import MovementSchedulerParams
 from nncf.torch.dynamic_graph.graph_tracer import ModelInputInfo

tests/torch/sparsity/movement/helpers/trainer.py

@@ -14,14 +14,14 @@
 
 import numpy as np
 import torch
+from datasets import Dataset  # pylint: disable=no-name-in-module
 from transformers import TrainingArguments
 from transformers.trainer import Trainer
 from transformers.trainer_callback import TrainerCallback
 from transformers.trainer_callback import TrainerControl
 from transformers.trainer_callback import TrainerState
 from transformers.trainer_utils import PREFIX_CHECKPOINT_DIR
 
-from datasets import Dataset  # pylint: disable=no-name-in-module
 from nncf.api.compression import CompressionAlgorithmController
 from nncf.common.compression import BaseCompressionAlgorithmController
 from nncf.common.utils.tensorboard import prepare_for_tensorboard

tests/torch/sparsity/movement/test_model_saving.py

@@ -18,6 +18,7 @@
 import pytest
 import torch
 from addict import Dict
+from datasets import Dataset  # pylint: disable=no-name-in-module
 from onnx import numpy_helper
 from openvino.runtime import Core
 from openvino.runtime import serialize
@@ -28,7 +29,6 @@
 from scipy.special import softmax
 from transformers.trainer_utils import PREFIX_CHECKPOINT_DIR
 
-from datasets import Dataset  # pylint: disable=no-name-in-module
 from nncf.torch import create_compressed_model
 from nncf.torch.checkpoint_loading import load_state
 from tests.torch.helpers import PTTensorListComparator

tests/torch/sparsity/movement/training_scripts/run_glue.py

@@ -12,13 +12,12 @@
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple
 
+import datasets
 import evaluate
 import jstyleson
 import numpy as np
 from transformers.training_args import ParallelMode
 
-import datasets
-
 # isort: off
 from nncf import NNCFConfig
 from nncf.api.compression import CompressionAlgorithmController