From 02d8588f5b2bac248aa482a32f7d0fbe8ad312d2 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Wed, 15 Apr 2020 09:19:49 -0500 Subject: [PATCH] bug fix: with column count --- pipeline.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/pipeline.py b/pipeline.py index 54e12c4..d218216 100644 --- a/pipeline.py +++ b/pipeline.py @@ -155,13 +155,13 @@ class Components : # df = reader() df = args['reader']() if 'reader' in args else args['data'] - if 'slice' in args and 'max_rows' in args['slice']: + # if 'slice' in args and 'max_rows' in args['slice']: - max_rows = args['slice']['max_rows'] - if df.shape[0] > max_rows : - print (".. slicing ") - i = np.random.choice(df.shape[0],max_rows,replace=False) - df = df.iloc[i] + # max_rows = args['slice']['max_rows'] + # if df.shape[0] > max_rows : + # print (".. slicing ") + # i = np.random.choice(df.shape[0],max_rows,replace=False) + # df = df.iloc[i]