From 65a1fadfcaeefc67df9b84ab5053103a72f218c6 Mon Sep 17 00:00:00 2001
From: Steve Nyemba <steve@the-phi.com>
Date: Sun, 12 Apr 2020 20:07:15 -0500
Subject: [PATCH] bug fix data type and pipeline

---
 data/gan.py | 3 ++-
 pipeline.py | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/data/gan.py b/data/gan.py
index a46740a..5975255 100644
--- a/data/gan.py
+++ b/data/gan.py
@@ -592,7 +592,8 @@ class Predict(GNet):
                                 # The code below will insure we have some acceptable cardinal relationships between id and synthetic values
                                 #
                                 
-                                df =   pd.DataFrame(np.round(f)).astype(np.int32)
+                                # df =   pd.DataFrame(np.round(f)).astype(np.int32)
+                                df =   pd.DataFrame(np.round(f),dtype=np.int32)
                                 
                                 p = 0 not in df.sum(axis=1).values
                                 x = df.sum(axis=1).values
diff --git a/pipeline.py b/pipeline.py
index 22c637d..c243ec3 100644
--- a/pipeline.py
+++ b/pipeline.py
@@ -125,9 +125,9 @@ class Components :
 		logger.write({"module":"train","action":"train","input":info})
 		data.maker.train(**_args)
 		
-		if set(['drone','autopilot']) in set( list(args.keys())) :
+		if 'autopilot' in ( list(args.keys())) :
 			print (['drone mode enabled ....'])
-			self.generate(**args)
+			self.generate(args)
 
 		pass