From 0656474ca890c62f30cef940a0e513afa8006db3 Mon Sep 17 00:00:00 2001
From: Steve Nyemba <steve@the-phi.com>
Date: Tue, 25 Feb 2020 12:27:45 -0600
Subject: [PATCH] bug fix: data-frame should be returned (not series)

---
 data/gan.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/data/gan.py b/data/gan.py
index 382dc41..f2f3cdb 100644
--- a/data/gan.py
+++ b/data/gan.py
@@ -624,7 +624,7 @@ class Predict(GNet):
                         
                 
                 tf.compat.v1.reset_default_graph()
-                
+                df = pd.DataFrame(df)
                 return df.to_dict(orient='list')
                         # return df.to_dict(orient='list')
                         # count = str(len(os.listdir(self.out_dir)))