def METHOD_NAME(self, preds, target, nan_strategy, nan_replace_value):
    """Test the differentiability of the metric, according to its `is_differentiable` attribute."""
    metric_args = {
        "nan_strategy": nan_strategy,
        "nan_replace_value": nan_replace_value,
        "num_classes": NUM_CLASSES,
    }
    self.run_differentiability_test(
        preds,
        target,
        metric_module=TheilsU,
        metric_functional=theils_u,
        metric_args=metric_args,
    )