mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-09 00:09:55 +00:00
Madhukar Mishra
37 lines
666 B
Python
37 lines
666 B
Python
import sys
|
|
|
|
import argbind
|
|
|
|
from dac.utils import download
|
|
from dac.utils.decode import decode
|
|
from dac.utils.encode import encode
|
|
|
|
STAGES = ["encode", "decode", "download"]
|
|
|
|
|
|
def run(stage: str):
|
|
"""Run stages.
|
|
|
|
Parameters
|
|
----------
|
|
stage : str
|
|
Stage to run
|
|
"""
|
|
if stage not in STAGES:
|
|
raise ValueError(f"Unknown command: {stage}. Allowed commands are {STAGES}")
|
|
stage_fn = globals()[stage]
|
|
|
|
if stage == "download":
|
|
stage_fn()
|
|
return
|
|
|
|
stage_fn()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
group = sys.argv.pop(1)
|
|
args = argbind.parse_args(group=group)
|
|
|
|
with argbind.scope(args):
|
|
run(group)
|