aou/data-maker
aou
/
data-maker
1
1
Fork 0
Code Issues 4 Pull Requests Packages Projects 1 Releases Wiki Activity

bugfix

This commit is contained in:
Steve Nyemba 2022-04-14 11:27:55 -05:00
parent 4345146f3a
commit 7a22314a46
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
data/maker/__init__.py
View File

@ -336,9 +336,10 @@ class factory :
:param autopilot will generate output automatically :param autopilot will generate output automatically
:param batch (default 2k) size of the batch :param batch (default 2k) size of the batch
""" """
if 'apply' not in _args :
return Trainer(**_args) if _args['apply'] == 'shuffe' :
elif _args['apply'] == 'shuffe' :
return Shuffle(**_args) return Shuffle(**_args)
elif _args['apply'] == 'generate' : elif _args['apply'] == 'generate' :
return Generator(**_args) return Generator(**_args)
else:
return Trainer(**_args)