diff --git a/pipeline.py b/pipeline.py index 2b5f028..4cac273 100644 --- a/pipeline.py +++ b/pipeline.py @@ -169,7 +169,7 @@ class Components : # let us fix the data types here every _id field will be an np.int64... # for name in df.columns.tolist(): - if name.endwith('_id') : + if name.endswith('_id') : df[name] = df[name].astype(np.int64)