...

4 years ago · 452014ec17
parent 7ccf9848b2
commit 452014ec17
1 changed files with 1 additions and 1 deletions
--- a/pipeline.py
+++ b/pipeline.py
@ -239,7 +239,7 @@ class Components :
 		real_df = pd.DataFrame()
 		if x_cols :
 			args['data'] = args['data'][list(set(args['data'].columns) - set(x_cols))]
-			real_df = args[x_cols].copy()
+			real_df = args['data'][x_cols].copy()
 		
 		args['candidates']	= 1 if 'candidates' not in args else int(args['candidates'])
 		if 'gpu' in args :