bugfix

2022-04-14 11:27:55 -05:00 · 2022-04-14 11:27:55 -05:00 · 7a22314a46
parent 4345146f3a
commit 7a22314a46
1 changed files with 5 additions and 4 deletions
--- 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)
+            return Generator(**_args)
+        else:
+            return Trainer(**_args)