- while result == None or max(result.values()) > 100:
- l = length
- if l > 5 and randomize_length:
- l = 5 + torch.randint(l - 5, (1,)).item()
- p, v = generate_program(nb_variables, l)
+ while result == None or max(result.values()) > result_max:
+ p, v = generate_program(nb_variables, operand_max, l)