fix error in pylint

apache · Aug 5, 2019 · d75e34b · d75e34b
1 parent eead5c2
commit d75e34b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/mxnet/ndarray/numpy/_op.py b/python/mxnet/ndarray/numpy/_op.py
@@ -667,7 +667,7 @@ def take(a, indices, axis=None, mode='clip', out=None):
     array([[4., 3.],
            [5., 7.]])
     """
-    if mode != 'wrap' and mode != 'clip':
+    if mode not in ('wrap', 'clip'):
         raise NotImplementedError(
             "function take does not support mode '{}'".format(mode))
     return _npi.take(a, indices, axis, mode, out)

diff --git a/python/mxnet/symbol/numpy/_symbol.py b/python/mxnet/symbol/numpy/_symbol.py
@@ -1313,7 +1313,7 @@ def take(a, indices, axis=None, mode='clip', out=None):
     - Only ndarray or scalar ndarray is accepted as valid input.
     - 'raise' mode is not supported.
     """
-    if mode != 'wrap' and mode != 'clip':
+    if mode not in ('wrap', 'clip'):
         raise NotImplementedError(
             "function take does not support mode '{}'".format(mode))
     return _npi.take(a, indices, axis, mode, out)