From ac8968c3e33cfe0408a67074bb7bbc0695981d21 Mon Sep 17 00:00:00 2001
From: Steve Nyemba <steve@the-phi.com>
Date: Tue, 30 Mar 2021 05:23:55 -0500
Subject: [PATCH] bug fix: fields skipped  (training)

---
 pipeline.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pipeline.py b/pipeline.py
index e643278..f4db40c 100644
--- a/pipeline.py
+++ b/pipeline.py
@@ -104,9 +104,6 @@ class Components :
 		
 		#
 		#
-		if 'ignore' in args and 'columns' in args['ignore'] :
-			_cols = self.get_ignore(data=df,columns=args['ignore']['columns'])
-			df = df[ list(set(df.columns)- set(_cols))]
 		# df = df.fillna('')
 		if schema :
 			_schema = []
@@ -125,6 +122,9 @@ class Components :
 		_args = copy.deepcopy(args)
 		# _args['store']  = args['store']['source']
 		_args['data'] = df
+		if 'ignore' in args and 'columns' in args['ignore'] :
+			_cols = self.get_ignore(data=df,columns=args['ignore']['columns'])
+			_args['data'] = df[ list(set(df.columns)- set(_cols))]
 		
 		data.maker.train(**_args)