mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
Added a way to run as a library by passing job dict
This commit is contained in:
13
run.py
13
run.py
@@ -1,5 +1,7 @@
|
||||
import os
|
||||
import sys
|
||||
from typing import Union, OrderedDict
|
||||
|
||||
sys.path.insert(0, os.getcwd())
|
||||
import argparse
|
||||
from toolkit.job import get_job
|
||||
@@ -19,6 +21,17 @@ def print_end_message(jobs_completed, jobs_failed):
|
||||
print("========================================")
|
||||
|
||||
|
||||
def run_job(
|
||||
config: Union[str, dict, OrderedDict],
|
||||
name=None
|
||||
):
|
||||
from toolkit.job import get_job
|
||||
|
||||
job = get_job(config, name)
|
||||
job.run()
|
||||
job.cleanup()
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user