o ©Ÿ?eã@s*dZddlZddlmZGdd„dƒZdS)z@Visitor restricting traversal to only the public tensorflow API.éN)Ú tf_inspectc@sReZdZdZdd„Zedd„ƒZedd„ƒZdd „Zdd d „Z d d„Z dd„Z d S)ÚPublicAPIVisitorzBVisitor to use with `traverse` to visit exactly the public TF API.cCs8||_d|_gd¢dgdœ|_gd¢dgdgdœ|_d S) zàConstructor. `visitor` should be a callable suitable as a visitor for `traverse`. It will be called only for members of the public TensorFlow API. Args: visitor: A visitor to call for the public API. Útf)ÚcompilerÚcoreÚsecurityZdtensorÚpythonZtslZ cpp_flags)rztf.flags)ZexamplesÚflagsÚplatformZpywrap_tensorflowZuser_opsZtoolsZ tensorboardr Zmock)rztf.appztf.testN)Ú_visitorÚ _root_nameÚ _private_mapÚ_do_not_descend_map)ÚselfÚvisitor©rúc/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/tools/common/public_api.pyÚ__init__s í îzPublicAPIVisitor.__init__cCó|jS)zÙA map from parents to symbols that should not be included at all. This map can be edited, but it should not be edited once traversal has begun. Returns: The map marking symbols to not include. )r ©rrrrÚ private_mapTó zPublicAPIVisitor.private_mapcCr)zØA map from parents to symbols that should not be descended into. This map can be edited, but it should not be edited once traversal has begun. Returns: The map marking symbols to not explore. ©rrrrrÚdo_not_descend_map`rz#PublicAPIVisitor.do_not_descend_mapcCs ||_dS)z'Override the default root name of 'tf'.N)r )rZ root_namerrrÚ set_root_namels zPublicAPIVisitor.set_root_nameNcCs:~||jvr ||j|vp| d¡rt d|¡ p|dvS)z!Return whether a name is private.Ú_z__.*__$)Ú__base__Ú __class__Z__next_in_mro__)r Ú startswithÚreÚmatch)rÚpathÚnameÚobjrrrÚ _is_privateps þzPublicAPIVisitor._is_privatecCs||jvo ||j|vS)zESafely queries if a specific fully qualified name should be excluded.r)rr!r"rrrÚ_do_not_descendxs  ÿz PublicAPIVisitor._do_not_descendcCs´t |¡rt| d¡ƒdkrtd|j|fƒ‚|r!d |j|g¡n|j}t|ƒD]\}}| |||¡r:|  ||f¡q(|  |||¡t|ƒD]\}}|  ||¡rW|  ||f¡qFdS)z.Visitor interface, see `traverse` for details.Ú.é zZModules nested too deep: %s.%s This is likely a problem with an accidental public import.N) rÚismoduleÚlenÚsplitÚ RuntimeErrorr ÚjoinÚlistr$Úremover r%)rr!ÚparentÚchildrenÚ full_pathr"ÚchildrrrÚ__call__}sþ€ €þzPublicAPIVisitor.__call__)N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚpropertyrrrr$r%r3rrrrrs;    r)r7rZtensorflow.python.utilrrrrrrÚs