INNER CODE UNIT · Python

generate_multi_scenes

allenai/Holodeck · ai2holodeck/main.py:80

def generate_multi_scenes(args):
    with open(args.query_file, "r") as f:
        queries = f.readlines()
        queries = [query.strip() for query in queries]

    for query in tqdm(queries):
        args.query = query
        generate_single_scene(args)


def generate_variants(args):
    try:
        original_scene = compress_json.load(args.original_scene)
    except:
        raise Exception(f"Could not load original scene from {args.original_scene}.")

    try:
        args.model.generate_variants(

View source record →