diff --git a/tests/ut/python/nn/test_loss.py b/tests/ut/python/nn/test_loss.py index 21e3006818..4139e32d74 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/onnx/test_onnx.py b/tests/ut/python/onnx/test_onnx.py index 29ed52ae28..a3de151ae9 100644 --- a/tests/ut/python/onnx/test_onnx.py +++ b/tests/ut/python/onnx/test_onnx.py @@ -17,7 +17,9 @@ import numpy as np import os import pytest import stat +import time +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/train/quant/test_quant.py b/tests/ut/python/train/quant/test_quant.py index 2424b7ce7c..e1acd0ccb8 100644 --- a/tests/ut/python/train/quant/test_quant.py +++ b/tests/ut/python/train/quant/test_quant.py @@ -17,6 +17,7 @@ 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 4c4ef2672a..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 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 f3919e6102..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 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 f8a9db3670..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