def METHOD_NAME(self, s_X):
    """Used internally by Lale for type-checking downstream operators."""
    if self._hyperparams["activate_flag"]:
        result = {
            "type": "array",
            "items": {"type": "array", "items": {"type": "number"}},
        }
    else:
        result = s_X
    return result