Fix lint in sparse_ndarray.py

eric-haibin-lin · Jun 12, 2017 · d7225ec · d7225ec
1 parent ab40b35
commit d7225ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/mxnet/sparse_ndarray.py b/python/mxnet/sparse_ndarray.py
@@ -397,7 +397,7 @@ class CSRNDArray(SparseNDArray):
 
     """
     def __reduce__(self):
-        return CSRNDArray, (None,), super(SparseNDArray, self).__getstate__()
+        return CSRNDArray, (None,), super(CSRNDArray, self).__getstate__()
 
     @property
     def indices(self):
@@ -440,7 +440,7 @@ class RowSparseNDArray(SparseNDArray):
     that have sparse gradients (e.g. SparseEmbedding).
     """
     def __reduce__(self):
-        return RowSparseNDArray, (None,), super(SparseNDArray, self).__getstate__()
+        return RowSparseNDArray, (None,), super(RowSparseNDArray, self).__getstate__()
 
     @property
     def indices(self):