diff --git a/data/maker/__init__.py b/data/maker/__init__.py index bce8d65..eb4c02d 100644 --- a/data/maker/__init__.py +++ b/data/maker/__init__.py @@ -336,9 +336,10 @@ class factory : :param autopilot will generate output automatically :param batch (default 2k) size of the batch """ - if 'apply' not in _args : - return Trainer(**_args) - elif _args['apply'] == 'shuffe' : + + if _args['apply'] == 'shuffe' : return Shuffle(**_args) elif _args['apply'] == 'generate' : - return Generator(**_args) \ No newline at end of file + return Generator(**_args) + else: + return Trainer(**_args) \ No newline at end of file