mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-21 23:09:15 +00:00
16 lines
417 B
Python
16 lines
417 B
Python
from jobs import BaseJob
|
|
from toolkit.config import get_config
|
|
|
|
|
|
def get_job(config_path) -> BaseJob:
|
|
config = get_config(config_path)
|
|
if not config['job']:
|
|
raise ValueError('config file is invalid. Missing "job" key')
|
|
|
|
job = config['job']
|
|
if job == 'extract':
|
|
from jobs import ExtractJob
|
|
return ExtractJob(config)
|
|
else:
|
|
raise ValueError(f'Unknown job type {job}')
|