diff --git a/tests/mindspore_test_framework/apps/bert_attention_submodules.py b/tests/mindspore_test_framework/apps/bert_attention_submodules.py index 7e49ce43d1..4ce72ffc84 100644 --- a/tests/mindspore_test_framework/apps/bert_attention_submodules.py +++ b/tests/mindspore_test_framework/apps/bert_attention_submodules.py @@ -167,7 +167,7 @@ class BertAttentionMask(nn.Cell): super(BertAttentionMask, self).__init__() self.has_attention_mask = has_attention_mask - self.multiply_data = Tensor([-1000.0, ], dtype=dtype) + self.multiply_data = Tensor([-1000.0,], dtype=dtype) self.multiply = P.Mul() if self.has_attention_mask: @@ -198,7 +198,7 @@ class BertAttentionMaskBackward(nn.Cell): dtype=mstype.float32): super(BertAttentionMaskBackward, self).__init__() self.has_attention_mask = has_attention_mask - self.multiply_data = Tensor([-1000.0, ], dtype=dtype) + self.multiply_data = Tensor([-1000.0,], dtype=dtype) self.multiply = P.Mul() self.attention_mask = Tensor(np.ones(shape=attention_mask_shape).astype(np.float32)) if self.has_attention_mask: diff --git a/tests/st/networks/test_gpu_lstm.py b/tests/st/networks/test_gpu_lstm.py index c85bcede80..bc59b7e387 100644 --- a/tests/st/networks/test_gpu_lstm.py +++ b/tests/st/networks/test_gpu_lstm.py @@ -136,7 +136,7 @@ def test_LSTM(): train_network.set_train() train_features = Tensor(np.ones([64, max_len]).astype(np.int32)) - train_labels = Tensor(np.ones([64, ]).astype(np.int32)[0:64]) + train_labels = Tensor(np.ones([64,]).astype(np.int32)[0:64]) losses = [] for epoch in range(num_epochs): loss = train_network(train_features, train_labels) diff --git a/tests/ut/python/exec/test_AssignAdd.py b/tests/ut/python/exec/test_AssignAdd.py index a7244c5639..502b7b92f1 100644 --- a/tests/ut/python/exec/test_AssignAdd.py +++ b/tests/ut/python/exec/test_AssignAdd.py @@ -45,7 +45,6 @@ class Net(nn.Cell): @non_graph_engine def test_AssignAdd_1(): """test AssignAdd 1""" - import mindspore.context as context context.set_context(mode=context.GRAPH_MODE) net = Net() x = Tensor(np.ones([1]).astype(np.int64) * 100) @@ -65,7 +64,6 @@ def test_AssignAdd_1(): @non_graph_engine def test_AssignAdd_2(): """test AssignAdd 2""" - import mindspore.context as context context.set_context(mode=context.GRAPH_MODE) net = Net() x = Tensor(np.ones([1]).astype(np.int64) * 102) diff --git a/tests/ut/python/ir/test_dtype.py b/tests/ut/python/ir/test_dtype.py index 704d09258f..1523a77ea3 100644 --- a/tests/ut/python/ir/test_dtype.py +++ b/tests/ut/python/ir/test_dtype.py @@ -13,9 +13,9 @@ # limitations under the License. # ============================================================================ """test_dtype""" +from dataclasses import dataclass import numpy as np import pytest -from dataclasses import dataclass import mindspore as ms from mindspore.common import dtype diff --git a/tests/ut/python/ir/test_tensor.py b/tests/ut/python/ir/test_tensor.py index e3e3347827..9cc7208d3f 100644 --- a/tests/ut/python/ir/test_tensor.py +++ b/tests/ut/python/ir/test_tensor.py @@ -34,7 +34,7 @@ ndarr = np.ones((2, 3)) def test_tensor_flatten(): with pytest.raises(AttributeError): - lst = [1, 2, 3, 4, ] + lst = [1, 2, 3, 4,] tensor_list = ms.Tensor(lst, ms.float32) tensor_list = tensor_list.Flatten() print(tensor_list) diff --git a/tests/ut/python/model/test_bert_cell.py b/tests/ut/python/model/test_bert_cell.py index 817a75b6b4..175341767a 100644 --- a/tests/ut/python/model/test_bert_cell.py +++ b/tests/ut/python/model/test_bert_cell.py @@ -107,7 +107,7 @@ class TrainStepWrapForAdamDynamicLr(nn.Cell): class TempC2Wrap(nn.Cell): - def __init__(self, op, c1=None, c2=None, ): + def __init__(self, op, c1=None, c2=None,): super(TempC2Wrap, self).__init__() self.op = op self.c1 = c1 @@ -387,7 +387,7 @@ test_case_cell_ops = [ 'block': set_train(nn.Dense(in_channels=768, out_channels=3072, activation='gelu', - weight_init=TruncatedNormal(0.02), )), + weight_init=TruncatedNormal(0.02),)), 'desc_inputs': [[3, 768]], 'desc_bprop': [[3, 3072]]}), ('GetNextSentenceOutput', { diff --git a/tests/ut/python/nn/optim/test_lars.py b/tests/ut/python/nn/optim/test_lars.py index c34033d884..d6fd4cd90e 100644 --- a/tests/ut/python/nn/optim/test_lars.py +++ b/tests/ut/python/nn/optim/test_lars.py @@ -12,8 +12,8 @@ # See the License for the specific language governing permissions and # limitations under the License. # ============================================================================ -import numpy as np from collections import Counter +import numpy as np import mindspore.nn as nn from mindspore import Tensor, Parameter diff --git a/tests/ut/python/nn/test_container.py b/tests/ut/python/nn/test_container.py index cd3e34a2fa..c381394c26 100644 --- a/tests/ut/python/nn/test_container.py +++ b/tests/ut/python/nn/test_container.py @@ -13,9 +13,9 @@ # limitations under the License. # ============================================================================ """ test container """ +from collections import OrderedDict import numpy as np import pytest -from collections import OrderedDict import mindspore.nn as nn from mindspore import Tensor diff --git a/tests/ut/python/nn/test_loss.py b/tests/ut/python/nn/test_loss.py index 4139e32d74..21e3006818 100644 --- a/tests/ut/python/nn/test_loss.py +++ b/tests/ut/python/nn/test_loss.py @@ -60,5 +60,5 @@ def test_SoftmaxCrossEntropyExpand(): loss = nn.SoftmaxCrossEntropyExpand() logits = Tensor(np.random.randint(0, 9, [100, 10]).astype(np.float32)) - labels = Tensor(np.random.randint(0, 9, [10, ]).astype(np.float32)) + labels = Tensor(np.random.randint(0, 9, [10,]).astype(np.float32)) _executor.compile(loss, logits, labels) diff --git a/tests/ut/python/nn/test_parameter.py b/tests/ut/python/nn/test_parameter.py index 886f855a79..3c66e0a6df 100644 --- a/tests/ut/python/nn/test_parameter.py +++ b/tests/ut/python/nn/test_parameter.py @@ -54,7 +54,6 @@ def test_parameter_tuple_illegal(): def test_parameter_init_illegal(): - import numpy as np dat = np.array([[1, 2, 3], [2, 3, 4]]) tensor = Tensor(dat) data_none = None diff --git a/tests/ut/python/onnx/test_onnx.py b/tests/ut/python/onnx/test_onnx.py index a3de151ae9..ca877548de 100644 --- a/tests/ut/python/onnx/test_onnx.py +++ b/tests/ut/python/onnx/test_onnx.py @@ -13,13 +13,11 @@ # limitations under the License. # ============================================================================ """ut for model serialize(save/load)""" -import numpy as np import os -import pytest import stat -import time +import numpy as np +import pytest -import mindspore.common.dtype as mstype import mindspore.nn as nn from mindspore import context from mindspore.common.parameter import Parameter diff --git a/tests/ut/python/predict/test_predict_save_model.py b/tests/ut/python/predict/test_predict_save_model.py index 20744eaf5a..5764678e19 100644 --- a/tests/ut/python/predict/test_predict_save_model.py +++ b/tests/ut/python/predict/test_predict_save_model.py @@ -20,8 +20,8 @@ Usage: """ import argparse -import numpy as np import os +import numpy as np import mindspore.context as context import mindspore.nn as nn diff --git a/tests/ut/python/train/quant/test_quant.py b/tests/ut/python/train/quant/test_quant.py index e1acd0ccb8..2424b7ce7c 100644 --- a/tests/ut/python/train/quant/test_quant.py +++ b/tests/ut/python/train/quant/test_quant.py @@ -17,7 +17,6 @@ import numpy as np from mobilenetv2_combined import MobileNetV2 import mindspore.context as context -import mindspore.ops.operations as P from mindspore import Tensor from mindspore import nn from mindspore.nn.layer import combined diff --git a/tests/ut/python/train/summary/test_graph_summary.py b/tests/ut/python/train/summary/test_graph_summary.py index 63ce4d3c74..79a56ba748 100644 --- a/tests/ut/python/train/summary/test_graph_summary.py +++ b/tests/ut/python/train/summary/test_graph_summary.py @@ -14,8 +14,8 @@ # ============================================================================ """ test_graph_summary """ import logging -import numpy as np import os +import numpy as np import mindspore.nn as nn from mindspore import Model, context diff --git a/tests/ut/python/train/summary/test_histogram_summary.py b/tests/ut/python/train/summary/test_histogram_summary.py index 47d463e2b3..dc0892167c 100644 --- a/tests/ut/python/train/summary/test_histogram_summary.py +++ b/tests/ut/python/train/summary/test_histogram_summary.py @@ -15,9 +15,9 @@ """Test histogram summary.""" import logging -import numpy as np import os import tempfile +import numpy as np from mindspore.common.tensor import Tensor from mindspore.train.summary._summary_adapter import _calc_histogram_bins diff --git a/tests/ut/python/train/summary/test_image_summary.py b/tests/ut/python/train/summary/test_image_summary.py index dba4134020..5e5bf2b3c3 100644 --- a/tests/ut/python/train/summary/test_image_summary.py +++ b/tests/ut/python/train/summary/test_image_summary.py @@ -19,8 +19,8 @@ @Desc : test summary function """ import logging -import numpy as np import os +import numpy as np import mindspore.nn as nn from mindspore import Model, context diff --git a/tests/ut/python/train/summary/test_summary.py b/tests/ut/python/train/summary/test_summary.py index a9cfb3521a..d26924a499 100644 --- a/tests/ut/python/train/summary/test_summary.py +++ b/tests/ut/python/train/summary/test_summary.py @@ -19,10 +19,11 @@ @Desc : test summary function """ import logging -import numpy as np import os -import pytest import random +import numpy as np +import pytest + import mindspore.nn as nn from mindspore.common.tensor import Tensor @@ -127,7 +128,7 @@ def test_scalar_summary_sample_with_shape_1(): class SummaryDemo(nn.Cell): """ SummaryDemo definition """ - def __init__(self, ): + def __init__(self,): super(SummaryDemo, self).__init__() self.s = P.ScalarSummary() self.histogram_summary = P.HistogramSummary() diff --git a/tests/ut/python/train/summary/test_summary_abnormal_input.py b/tests/ut/python/train/summary/test_summary_abnormal_input.py index ad206c65d3..95cbf7549b 100644 --- a/tests/ut/python/train/summary/test_summary_abnormal_input.py +++ b/tests/ut/python/train/summary/test_summary_abnormal_input.py @@ -19,8 +19,8 @@ @Desc : test summary function of abnormal input """ import logging -import numpy as np import os +import numpy as np from mindspore.common.tensor import Tensor from mindspore.train.summary.summary_record import SummaryRecord diff --git a/tests/ut/python/train/summary/test_tensor_summary.py b/tests/ut/python/train/summary/test_tensor_summary.py index bb41951aa2..9a028a8a12 100644 --- a/tests/ut/python/train/summary/test_tensor_summary.py +++ b/tests/ut/python/train/summary/test_tensor_summary.py @@ -19,8 +19,8 @@ @Desc : test summary function """ import logging -import numpy as np import os +import numpy as np import mindspore.nn as nn from mindspore.common.tensor import Tensor @@ -110,7 +110,7 @@ def get_test_data_check(step): class SummaryDemo(nn.Cell): """ SummaryDemo definition """ - def __init__(self, ): + def __init__(self,): super(SummaryDemo, self).__init__() self.s = P.TensorSummary() self.add = P.TensorAdd() diff --git a/tests/ut/python/utils/test_callback.py b/tests/ut/python/utils/test_callback.py index 04db64649a..e9f8c4e708 100644 --- a/tests/ut/python/utils/test_callback.py +++ b/tests/ut/python/utils/test_callback.py @@ -13,10 +13,10 @@ # limitations under the License. # ============================================================================ """test callback function.""" -import numpy as np import os -import pytest import stat +import numpy as np +import pytest import mindspore.common.dtype as mstype import mindspore.nn as nn diff --git a/tests/ut/python/utils/test_initializer.py b/tests/ut/python/utils/test_initializer.py index 15d8838a46..417d0bb2b1 100644 --- a/tests/ut/python/utils/test_initializer.py +++ b/tests/ut/python/utils/test_initializer.py @@ -14,9 +14,9 @@ # ============================================================================ """ test_initializer """ import math +from functools import reduce import numpy as np import pytest as py -from functools import reduce from scipy import stats import mindspore as ms diff --git a/tests/ut/python/utils/test_serialize.py b/tests/ut/python/utils/test_serialize.py index 5c0eb6f8dd..a248a49bce 100644 --- a/tests/ut/python/utils/test_serialize.py +++ b/tests/ut/python/utils/test_serialize.py @@ -13,11 +13,11 @@ # limitations under the License. # ============================================================================ """ut for model serialize(save/load)""" -import numpy as np import os -import pytest import stat import time +import pytest +import numpy as np import mindspore.common.dtype as mstype import mindspore.nn as nn @@ -31,7 +31,7 @@ from mindspore.ops import operations as P from mindspore.train.callback import _CheckpointManager from mindspore.train.serialization import save_checkpoint, load_checkpoint, load_param_into_net, \ _exec_save_checkpoint, export, _save_graph -from ..ut_filter import run_on_onnxruntime, non_graph_engine +from ..ut_filter import non_graph_engine context.set_context(mode=context.GRAPH_MODE)