Remove self from the default session stack.
(self)
| 821 | self._default_session.__enter__() |
| 822 | |
| 823 | def _unregister_default(self): |
| 824 | """Remove self from the default session stack.""" |
| 825 | if self._default_session: |
| 826 | self._default_session.__exit__(None, None, None) |
| 827 | self._default_session = None |
| 828 | |
| 829 | def _wrapper(self, dag_node: DAGNode) -> Union[DAGNode, App, Context, Graph, Any]: |
| 830 | if self.eager(): |