INNER CODE UNIT · Python

generate_variants

allenai/Holodeck · ai2holodeck/main.py:90

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(
            query=args.query,
            original_scene=original_scene,
            save_dir=args.save_dir,
            number_of_variants=int(args.number_of_variants),
            used_assets=args.used_assets,
        )
    except:
        print(f"Could not generate variants from {args.query}.")

View source record →