keras/src/utils/backend_utils.py/TFGraphScope/__exit__
class TFGraphScope:    def __exit__(self, *args, **kwargs):
        global_state.set_global_attribute(
            "in_tf_graph_scope", self._original_value
        )