BVh|dZddlmZmZddlZddlmZmZddl m Z edZ dedee fd edee ffd Z y) z9TFDecorator-aware replacements for the contextlib module.)CallableIteratorN)ContextManagerTypeVar) tf_decorator_Ttarget.returncZtj|}tj||dS)aA tf_decorator-aware wrapper for `contextlib.contextmanager`. Usage is identical to `contextlib.contextmanager`. Args: target: A callable to be wrapped in a contextmanager. Returns: A callable that can be used inside of a `with` statement. contextmanager) _contextlibr rmake_decorator)r context_managers T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/util/tf_contextlib.pyr r s* ..v6/  $ $V_>N OO) __doc__collections.abcrr contextlibr typingrrtensorflow.python.utilrrr rrrsU@. */ T] P S(2,& ' P c>"%%& Pr