Jack

You can wrap kendalls_tau with [tf.py_function](https://www.tensorflow.org/api_docs/python/tf/py_function): ```python def tf_kendalls_tau(y_true, y_pred): kt = tf.py_function( kendalls_tau, (y_true, y_pred), tf.float32 ) return kt model.compile(optimizer="Adam", loss="mse", metrics=tf_kendalls_tau) ``` The error is caused by calling...

Thank you for following up. I'd love to help out on this. Let me know if there's anything I can do that would be valuable.