wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-06-27 17:47:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

BUG: fix sanitize_df to min of int32

Browse Source
This commit is contained in:
Victor Grau Serrat
2018-03-12 16:52:54 -06:00
committed by lacabra
parent 437e5c5b80
commit 60e07c8a3c
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
catalyst/marketplace/utils/bundle_utils.py
+2
View File
@@ -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)