bug fix

3 years ago · 7de89a576a
parent b10296246d
commit 7de89a576a
1 changed files with 5 additions and 1 deletions
--- 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
    #