from core.imitation_learning import ImitationLearning
from core.config import config
import argparse


def get_arguments():
    parser = argparse.ArgumentParser()
    parser.add_argument("-t", "--task_name",
                        help="Task name.",
                        default="reach_target")
    parser.add_argument("-n", "--n_samples",
                        help="Task name.",
                        type=int,
                        default=200)
    args = parser.parse_args()

    return args


if __name__== "__main__":
    args = get_arguments()
    ImitationLearning(config[args.task_name]["task_class"], config[args.task_name]["n_features"], True, args.n_samples)
c856ab54fc506101b660c4dfe1aa0cdefc945e0d
<mask token>

<mask token>


def get_arguments():
    parser = argparse.ArgumentParser()
    parser.add_argument('-t', '--task_name', help='Task name.', default=
        'reach_target')
    parser.add_argument('-n', '--n_samples', help='Task name.', type=int,
        default=200)
    args = parser.parse_args()
    return args


<mask token>

<mask token>


def get_arguments():
    parser = argparse.ArgumentParser()
    parser.add_argument('-t', '--task_name', help='Task name.', default=
        'reach_target')
    parser.add_argument('-n', '--n_samples', help='Task name.', type=int,
        default=200)
    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = get_arguments()
    ImitationLearning(config[args.task_name]['task_class'], config[args.
        task_name]['n_features'], True, args.n_samples)

from core.imitation_learning import ImitationLearning
from core.config import config
import argparse


def get_arguments():
    parser = argparse.ArgumentParser()
    parser.add_argument('-t', '--task_name', help='Task name.', default=
        'reach_target')
    parser.add_argument('-n', '--n_samples', help='Task name.', type=int,
        default=200)
    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = get_arguments()
    ImitationLearning(config[args.task_name]['task_class'], config[args.
        task_name]['n_features'], True, args.n_samples)

from core.imitation_learning import ImitationLearning
from core.config import config
import argparse


def get_arguments():
    parser = argparse.ArgumentParser()
    parser.add_argument("-t", "--task_name",
                        help="Task name.",
                        default="reach_target")
    parser.add_argument("-n", "--n_samples",
                        help="Task name.",
                        type=int,
                        default=200)
    args = parser.parse_args()

    return args


if __name__== "__main__":
    args = get_arguments()
    ImitationLearning(config[args.task_name]["task_class"], config[args.task_name]["n_features"], True, args.n_samples)