Fix TypeError

✅ Solution

TypeError in TensorFlow, particularly with `jit_compile=True`, often arises when a Python boolean or a function not decorated with `tf.function` is used as a conditional predicate within TensorFlow control flow operations like `tf.cond`. To fix this, ensure that the conditional predicate is a TensorFlow boolean tensor (e.g., created using `tf.constant(True, dtype=tf.bool)`) or a callable that returns a TensorFlow boolean tensor, and decorate the conditional branch functions with `@tf.function` for graph compilation. Convert any Python bool to tf.constant(bool_value, dtype=tf.bool) before using it.