@@ -64,33 +64,32 @@ def serialize(self, data):
64
64
file, or buffer.
65
65
66
66
Returns:
67
- _io .BytesIO: A buffer containing data serialzied in the .npy format.
67
+ io .BytesIO: A buffer containing data serialzied in the .npy format.
68
68
"""
69
69
if isinstance (data , np .ndarray ):
70
70
if not data .size > 0 :
71
71
raise ValueError ("Cannot serialize empty array." )
72
- return self .serialize_array (data )
72
+ return self ._serialize_array (data )
73
73
74
74
if isinstance (data , list ):
75
75
if not len (data ) > 0 :
76
- raise ValueError ("empty array can't be serialized " )
77
- return self .serialize_array (np .array (data , self .dtype ))
76
+ raise ValueError ("Cannot serialize empty array. " )
77
+ return self ._serialize_array (np .array (data , self .dtype ))
78
78
79
79
# files and buffers. Assumed to hold npy-formatted data.
80
80
if hasattr (data , "read" ):
81
81
return data .read ()
82
82
83
- return self .serialize_array (np .array (data ))
83
+ return self ._serialize_array (np .array (data ))
84
84
85
- @staticmethod
86
- def serialize_array (array ):
85
+ def _serialize_array (self , array ):
87
86
"""Saves a NumPy array in a buffer.
88
87
89
88
Args:
90
89
array (numpy.ndarray): The array to serialize.
91
90
92
91
Returns:
93
- _io .BytesIO: A buffer containing the serialized array.
92
+ io .BytesIO: A buffer containing the serialized array.
94
93
"""
95
94
buffer = io .BytesIO ()
96
95
np .save (buffer , array )
0 commit comments