From 6bbe7677146b7914ff8ffa37dd44bfbce96837de Mon Sep 17 00:00:00 2001
From: Steve Nyemba <steve@the-phi.com>
Date: Thu, 1 Apr 2021 14:13:00 -0500
Subject: [PATCH] ...

---
 pipeline.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pipeline.py b/pipeline.py
index 29d15a7..6476221 100644
--- 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 = df['data'][x_cols].copy()
+			real_df = df[x_cols].copy()
 		
 		args['candidates']	= 1 if 'candidates' not in args else int(args['candidates'])
 		if 'gpu' in args :
@@ -279,7 +279,7 @@ class Components :
 				if x_cols :
 					for _col in x_cols :
 						if real_df[_col].unique().size > 0 :
-							_df[_col] = self.approximate(df[_col].fillna(-1))
+							_df[_col] = self.approximate(real_df[_col].fillna(-1))
 						else:
 							_df[_col] = -1