TypeError: train_eval_model() got an unexpected keyword argument 'use_tpu_wrapper'

[aRnaZ91]

Hi,
I have tried to run QT-opt from my PC last week but I have not been able to make it work yet.
I am using conda to be able to get the right version of every packages needed to run the project (and especially to get tensorflow 1 and avoid the migration to tensorflow 2). Though, even if I have the packages of requirement.txt with the right versions, I am still getting errors like showed on the screenshot (problem with use_tpu_wrapper). Do you have any ideas of how to fix these errors to run the code in the right way?
Thank you in advance for your consideration.

[anandhupvr]

Hi,
by removing or commenting use_tpu_wrapper=params['use_tpu_wrapper'] argument from t2r_test_fixture.py solve this error.

inside random_train_model function
train_eval.train_eval_model( t2r_model=tf_model, input_generator_train=input_generator, max_train_steps=params['max_train_steps'], model_dir=params['model_dir'], use_tpu_wrapper=params['use_tpu_wrapper'])