aou/data-maker
aou
/
data-maker
1
1
Fork 0
Code Issues 4 Pull Requests Packages Projects 1 Releases Wiki Activity

bug fix with typing

This commit is contained in:
Steve Nyemba 2020-04-09 19:21:25 -05:00
parent 944a3edbf6
commit e4b164a34b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pipeline.py
View File

@ -168,8 +168,13 @@ class Components :
# How many rows sub-partition must we divide this into ? # How many rows sub-partition must we divide this into ?
# let us fix the data types here every _id field will be an np.int64... # let us fix the data types here every _id field will be an np.int64...
# #
for name in df.columns.tolist(): for name in df.columns.tolist():
if name.endswith('_id') : if name.endswith('_id') :
if df[name].isnull().sum() > 0 :
df[name].fillna(0,inplace=True)
else:
df[name] = df[name].astype(np.int64) df[name] = df[name].astype(np.int64)