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 #