From 732ccb42e5cd886da632f5a0f523db13bbc07494 Mon Sep 17 00:00:00 2001
From: Steve Nyemba <steve@the-phi.com>
Date: Thu, 1 Apr 2021 12:43:09 -0500
Subject: [PATCH] ...

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

diff --git a/pipeline.py b/pipeline.py
index 6f28eac..8b1dd9e 100644
--- a/pipeline.py
+++ b/pipeline.py
@@ -136,6 +136,8 @@ class Components :
 		# We need to make sure that continuous columns are removed 
 		if x_cols :
 			_args['data'] = df[list(set(df.columns) - set(x_cols))]
+		if 'gpu' in args :
+			_args['gpu'] = args['gpu']
 		data.maker.train(**_args)
 		
 		if 'autopilot' in ( list(args.keys())) :