Skip to content

Commit 00da404

Browse files
committed
better naming && fixed docstring
1 parent 572b0f2 commit 00da404

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/amazon/common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,8 @@ def write_numpy_to_dense_tensor(file, array, labels=None):
108108
_write_recordio(file, record.SerializeToString())
109109

110110

111-
def write_numpy_to_sparse_tensor(file, array, labels=None):
112-
"""Writes a numpy array to a dense tensor"""
111+
def write_spmatrix_to_sparse_tensor(file, array, labels=None):
112+
"""Writes a scipy sparse matrix to a sparse tensor"""
113113

114114
if not issparse(array):
115115
raise TypeError("Array must be sparse")

0 commit comments

Comments
 (0)