Extract docs from tf namespace and write them to disk.
(py_modules,
private_map,
do_not_descend_map,
visitor_cls=doc_generator_visitor.DocGeneratorVisitor)
| 290 | |
| 291 | |
| 292 | def extract(py_modules, |
| 293 | private_map, |
| 294 | do_not_descend_map, |
| 295 | visitor_cls=doc_generator_visitor.DocGeneratorVisitor): |
| 296 | """Extract docs from tf namespace and write them to disk.""" |
| 297 | # Traverse the first module. |
| 298 | visitor = visitor_cls(py_modules[0][0]) |
| 299 | api_visitor = DocControlsAwareCrawler(visitor) |
| 300 | api_visitor.set_root_name(py_modules[0][0]) |
| 301 | add_dict_to_dict(private_map, api_visitor.private_map) |
| 302 | add_dict_to_dict(do_not_descend_map, api_visitor.do_not_descend_map) |
| 303 | |
| 304 | traverse.traverse(py_modules[0][1], api_visitor) |
| 305 | |
| 306 | # Traverse all py_modules after the first: |
| 307 | for module_name, module in py_modules[1:]: |
| 308 | visitor.set_root_name(module_name) |
| 309 | api_visitor.set_root_name(module_name) |
| 310 | traverse.traverse(module, api_visitor) |
| 311 | |
| 312 | return visitor |
| 313 | |
| 314 | |
| 315 | class _GetMarkdownTitle(py_guide_parser.PyGuideParser): |
no test coverage detected