def eval_step(self, model: Model, inputs):
"""Run one evaluation step"""
return model.eval_step(inputs)
def eval_step(self, model: Model, inputs):
"""Run one evaluation step"""
outputs = model.eval_step(inputs)
outputs = {k: v.tolist()[0] for k, v in outputs.items()}
return outputs