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

---
 pipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pipeline.py b/pipeline.py
index 7082b71..29d15a7 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 = args['data'][x_cols].copy()
+			real_df = df['data'][x_cols].copy()
 		
 		args['candidates']	= 1 if 'candidates' not in args else int(args['candidates'])
 		if 'gpu' in args :