Compare with TF-Agents