Hi Ed,
As a sanity check, could you provide the TFMA version in your image? The TFX evaluator uses
TFMA as the underlying module, and TF-DF support was added in TFMA 0.34.0 [github]. For reference, TFX 1.5.0 uses TFMA 0.36.0.
Please also check the worker log of the dataflow job created by the TFX evaluator (Dataflow → Jobs → click on the job → click ‘Show’ at the bottom next to Logs → WORKER LOGS → enter ‘tensorflow_decision_forests’ in filter), and find the line imported tensorflow_decision_forests or tensorflow_decision_forests is not available: No module named 'tensorflow_decision_forests'. Please let us know whether TF-DF is properly imported.
If everything checks out but the problem persists, we can try a more general workaround for custom ops.
Best,
Alister