Browse Source

!1533 Fix pylint warning for pynative test cases

Merge pull request !1533 from seatea/fix-pylint-warning-for-pynative-mode
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
ff4e4c2ee2
3 changed files with 5 additions and 5 deletions
  1. +1
    -1
      tests/ut/python/pynative_mode/test_backend.py
  2. +1
    -1
      tests/ut/python/pynative_mode/test_bprop.py
  3. +3
    -3
      tests/ut/python/pynative_mode/test_cell_bprop.py

+ 1
- 1
tests/ut/python/pynative_mode/test_backend.py View File

@@ -24,7 +24,7 @@ from mindspore.common.parameter import Parameter
from mindspore.ops import operations as P


def setup_module(module):
def setup_module():
context.set_context(mode=context.PYNATIVE_MODE)




+ 1
- 1
tests/ut/python/pynative_mode/test_bprop.py View File

@@ -24,7 +24,7 @@ from mindspore.ops import operations as P
from ....mindspore_test_framework.utils.bprop_util import bprop


def setup_module(module):
def setup_module():
context.set_context(mode=context.PYNATIVE_MODE)




+ 3
- 3
tests/ut/python/pynative_mode/test_cell_bprop.py View File

@@ -219,7 +219,7 @@ class OneInputBprop(nn.Cell):
return self.op(x)

def bprop(self, x, out, dout):
return 5 * x,
return (5 * x,)


def test_grad_one_input_bprop():
@@ -349,7 +349,7 @@ class MulAddWithWrongOutputNum(nn.Cell):
return 2 * x + y

def bprop(self, x, y, out, dout):
return 2 * dout,
return (2 * dout,)


def test_grad_mul_add_with_wrong_output_num():
@@ -380,7 +380,7 @@ def test_grad_mul_add_with_wrong_output_type():
class MulAddWithWrongOutputShape(nn.Cell):
def __init__(self):
super(MulAddWithWrongOutputShape, self).__init__()
self.ones = Tensor(np.ones([2, ]))
self.ones = Tensor(np.ones([2,]))

def construct(self, x, y):
return 2 * x + y


Loading…
Cancel
Save