# Path: studio/runner.py
# def main(args=sys.argv[1:]):
# def spin_up_workers(
#         runner_args,
#         config,
#         resources_needed,
#         queue_name,
#         verbose=10):
# def get_experiment_fitnesses(experiments, optimizer, config, logger):
# def _parse_artifacts(art_list, mutable):
# def _parse_external_artifacts(art_list, db):
# def _setup_artifacts_creds(artifacts, storage_creds):
# def _parse_hardware(runner_args, config={}):
# def _add_hyperparam_experiments(
#         exec_filename,
#         other_args,
#         runner_args,
#         artifacts,
#         resources_needed,
#         logger,
#         optimizer=None,
#         hyperparam_tuples=None):
#     def create_experiments(hyperparam_tuples):


import sys
import argparse
from studio import runner


def main():
    argparser = argparse.ArgumentParser()
    argparser.add_argument(
        '--wrapper', '-w',
        help='python script with function create_model ' +
             'that takes modeldir '
        '(that is, directory where experiment saves ' +
        'the checkpoints etc)' +
        'and returns dict -> dict function (model).' +
        'By default, studio-serve will try to determine ' +
        'this function automatically.',
        default=None
    )

    argparser.add_argument('--port',

# Path: studio/runner.py
# def main(args=sys.argv[1:]):
# def spin_up_workers(
#         runner_args,
#         config,
#         resources_needed,
#         queue_name,
#         verbose=10):
# def get_experiment_fitnesses(experiments, optimizer, config, logger):
# def _parse_artifacts(art_list, mutable):
# def _parse_external_artifacts(art_list, db):
# def _setup_artifacts_creds(artifacts, storage_creds):
# def _parse_hardware(runner_args, config={}):
# def _add_hyperparam_experiments(
#         exec_filename,
#         other_args,
#         runner_args,
#         artifacts,
#         resources_needed,
#         logger,
#         optimizer=None,
#         hyperparam_tuples=None):
#     def create_experiments(hyperparam_tuples):


# Path: studio/serve.py
import sys
import argparse
from studio import runner



def main():
    argparser = argparse.ArgumentParser()
    argparser.add_argument(
        '--wrapper', '-w',
        help='python script with function create_model ' +
             'that takes modeldir '
        '(that is, directory where experiment saves ' +
        'the checkpoints etc)' +
        'and returns dict -> dict function (model).' +
        'By default, studio-serve will try to determine ' +
        'this function automatically.',
        default=None
    )

    argparser.add_argument('--port',

                           help='port to run Flask server on',