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)