..

pipeline.py

@@ -274,10 +274,6 @@ class Components :
 				#
 				
 				_df = _df[list(set(_df.columns)  - set(skip_columns))]
-				
-				if set(df.columns) & set(_df.columns) :
-					_columns = set(df.columns) - set(_df.columns)
-					df = df[_columns]
 				if x_cols :
 					for _col in x_cols :
 						if df[_col].unique().size > 0 :
@@ -285,6 +281,10 @@ class Components :
 						else:
 							_df[_col] = -1
 				
+				if set(df.columns) & set(_df.columns) :
+					_columns = set(df.columns) - set(_df.columns)											
+					df = df[_columns]
+
 				#
 				# Let us merge the dataset here and and have a comprehensive dataset