From 7a22314a46ca5c806428f63cb3f90419e99b1af1 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Thu, 14 Apr 2022 11:27:55 -0500 Subject: [PATCH] bugfix --- data/maker/__init__.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) 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