Make record serializer accept one-dimensional arrays (#320) (#334)

Author: jnclt

CHANGELOG.rst

@@ -2,6 +2,11 @@
 CHANGELOG
 =========
 
+1.9.1dev
+========
+
+* bug-fix: Estimators: Fix serialization of single records
+
 1.9.0
 =====
 

src/sagemaker/amazon/common.py

@@ -29,7 +29,7 @@ def __init__(self, content_type='application/x-recordio-protobuf'):
 
     def __call__(self, array):
         if len(array.shape) == 1:
-            array.reshape(1, array.shape[0])
+            array = array.reshape(1, array.shape[0])
         assert len(array.shape) == 2, "Expecting a 1 or 2 dimensional array"
         buf = io.BytesIO()
         write_numpy_to_dense_tensor(buf, array)

tests/unit/test_common.py

@@ -32,6 +32,16 @@ def test_serializer():
         assert record.features["values"].float64_tensor.values == expected
 
 
+def test_serializer_accepts_one_dimensional_array():
+    s = numpy_to_record_serializer()
+    array_data = [1.0, 2.0, 3.0]
+    buf = s(np.array(array_data))
+    record_data = next(_read_recordio(buf))
+    record = Record()
+    record.ParseFromString(record_data)
+    assert record.features["values"].float64_tensor.values == array_data
+
+
 def test_deserializer():
     array_data = [[1.0, 2.0, 3.0], [10.0, 20.0, 30.0]]
     s = numpy_to_record_serializer()