+ if args.sandbox_level == 0:
+ problem = tasks.ProblemLevel0(
+ nb_sentences=args.sandbox_levels_nb_items,
+ len_prompt=args.sandbox_levels_len_source,
+ len_result=args.sandbox_levels_len_result,
+ )
+ elif args.sandbox_level == 1:
+ problem = tasks.ProblemLevel1(
+ nb_operators=args.sandbox_levels_nb_items,
+ len_source=args.sandbox_levels_len_source,
+ len_result=args.sandbox_levels_len_result,
+ )
+ elif args.sandbox_level == 2:
+ problem = tasks.ProblemLevel2(
+ len_source=args.sandbox_levels_len_source,
+ len_result=args.sandbox_levels_len_result,
+ )
+ else:
+ raise ValueError(f"Unknown sandbox level {args.sandbox_level}")
+