ape/ape.py

# -*-coding:utf-8 -*-

from ape.instance import Instance, LoadFactory
from functools import partial


def load_task(task, file):
    global instance
    if task:
        loader = LoadFactory[task]
        print(loader)
    else:
        print(file)
        loader = partial(LoadFactory['upload'], file=[i.name for i in file][0])
    instance = Instance.from_file(loader)
    print(instance.samples[0])
    return instance


def sample_data(instance, n_train, n_few_shot, n_eval):
    instance.sample(n_train, n_few_shot, n_eval)
    train_str = instance.display(instance.train_samples)
    eval_str = instance.display(instance.eval_samples)
    return train_str, eval_str, instance