Yunfeng Zhang

I tried setting Dataflow's max_num_workers to 1 and the job succeeded. Looks like the problem is indeed in running dataflow with multiple workers.

Hi @singhniraj08, I'm using the official TFX Evaluator, which internally uses tfma.ExtractEvaluateAndWriteResults as you can see [here](https://github.com/tensorflow/tfx/blob/r1.2.0/tfx/components/evaluator/executor.py#L285).

Thank you @kumarpiyush for the information. I tried your suggestion but I get a different error: `IndexError: arrays used as indices must be of integer (or boolean) type [while running...

@DirkjanVerdoorn Thanks for the suggestion. The labels are integer types so that's really not the problem.

Just to follow up on this, now our environment is updated to TFMA 0.31 and I can confirm that the fairness_indicator_thresholds parameter in Evaluator still doesn't work, but @kumarpiyush's method...