commit 3a99698734b3230b09fa641c98d3e17a973b50b1
author Bram Wasti <bwasti@fb.com> Tue Aug 01 13:44:48 2017 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Tue Aug 01 13:53:11 2017 -0700
tree 5cdd903fac5cd50349a5c8e8d839a5d77366d9fe
parent 5d304a3b49d0d0781538c260897c8c2377775c6c

include numpy's other 32bit int type

Summary: forgot one :)

Reviewed By: akyrola

Differential Revision: D5534905

fbshipit-source-id: a0e58ca3922ec80f526f7586931ff3da8e9bcffc

caffe2/python/utils.py

diff --git a/caffe2/python/utils.py b/caffe2/python/utils.py
index 27560b4..a5afaf2 100644
--- a/caffe2/python/utils.py
+++ b/caffe2/python/utils.py
@@ -65,7 +65,7 @@
     elif arr.dtype == np.float64:
         tensor.data_type = caffe2_pb2.TensorProto.DOUBLE
         tensor.double_data.extend(list(arr.flatten().astype(np.float64)))
-    elif arr.dtype == np.int:
+    elif arr.dtype == np.int or arr.dtype == np.int32:
         tensor.data_type = caffe2_pb2.TensorProto.INT32
         tensor.int32_data.extend(list(arr.flatten().astype(np.int)))
     elif arr.dtype == np.int16: