From 60e07c8a3c31938a325065be38eeebb5ffde74f5 Mon Sep 17 00:00:00 2001 From: Victor Grau Serrat Date: Mon, 12 Mar 2018 16:52:54 -0600 Subject: [PATCH] BUG: fix sanitize_df to min of int32 --- catalyst/marketplace/utils/bundle_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/catalyst/marketplace/utils/bundle_utils.py b/catalyst/marketplace/utils/bundle_utils.py index e489c0b5..5a0b2f6c 100644 --- a/catalyst/marketplace/utils/bundle_utils.py +++ b/catalyst/marketplace/utils/bundle_utils.py @@ -88,5 +88,7 @@ def safely_reduce_dtype(ser): # pandas.Series or numpy.array new_itemsize = np.min_scalar_type(val).itemsize if mx < new_itemsize: mx = new_itemsize + if orig_dtype == 'int': + mx = max(mx, 4) new_dtype = orig_dtype + str(mx * 8) return ser.astype(new_dtype)