def imprint(env, input_arr):
 env.generate_input_data = (lambda _: input_arr)

def imprint ( env , input_arr ) : env . generate_input_data = ( lambda _ : input_arr )
monkey-patch the given environment so that when reset() is called .