From 7de89a576ae0595ccc5eca2264bbc85d8a34afc1 Mon Sep 17 00:00:00 2001
From: Steve Nyemba <steve@the-phi.com>
Date: Mon, 10 May 2021 14:43:29 -0500
Subject: [PATCH] bug fix

---
 data/maker/__init__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/data/maker/__init__.py b/data/maker/__init__.py
index 4867bf6..3a4caf6 100644
--- a/data/maker/__init__.py
+++ b/data/maker/__init__.py
@@ -174,7 +174,11 @@ def generate(**_args):
         args['gpu'] = _args['gpu']
        
     handler     = gan.Predict (**args)
-    handler.load_meta(column=None)
+    lparams = {'columns':None}
+    if partition :
+        lparams['partition'] = partition
+    
+    handler.load_meta(lparams)
     #
     # Let us now format the matrices by reverting them to a data-frame with values
     #