File tree Expand file tree Collapse file tree 3 files changed +38
-6
lines changed
tools/tensorflow_docs/api_generator Expand file tree Collapse file tree 3 files changed +38
-6
lines changed Original file line number Diff line number Diff line change @@ -35,20 +35,25 @@ def is_deprecated(obj) -> bool:
35
35
36
36
def inheritable_header (text : str ):
37
37
38
- def _wrapped (cls ):
39
- setattr (cls , _INHERITABLE_HEADER , text )
40
- return cls
38
+ def _wrapped (obj ):
39
+ setattr (obj , _INHERITABLE_HEADER , text )
40
+ return obj
41
41
42
42
return _wrapped
43
43
44
44
45
- def get_inheritable_header (cls ) -> Optional [str ]:
46
- return getattr (cls , _INHERITABLE_HEADER , None )
45
+ def get_inheritable_header (obj ) -> Optional [str ]:
46
+ return getattr (obj , _INHERITABLE_HEADER , None )
47
+
48
+
49
+ header = inheritable_header
50
+ get_header = get_inheritable_header
47
51
48
52
49
53
_NO_SEARCH_HINTS = "_tf_docs_no_search_hints"
50
54
51
55
56
+
52
57
def hide_from_search (obj : T ) -> T :
53
58
"""Marks an object so metadata search hints will not be included on it's page.
54
59
Original file line number Diff line number Diff line change 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
# ==============================================================================
15
- from typing import Any
15
+ import textwrap
16
+ from typing import Any , Optional
16
17
18
+ from tensorflow_docs .api_generator import doc_controls
17
19
from tensorflow_docs .api_generator import parser
18
20
from tensorflow_docs .api_generator import signature as signature_lib
19
21
from tensorflow_docs .api_generator .pretty_docs import base_page
@@ -86,5 +88,12 @@ def decorators(self):
86
88
def add_decorator (self , dec ):
87
89
self ._decorators .append (dec )
88
90
91
+ @property
92
+ def header (self ) -> Optional [str ]:
93
+ header = doc_controls .get_header (self .py_object )
94
+ if header is not None :
95
+ header = textwrap .dedent (header )
96
+ return header
97
+
89
98
def get_metadata_html (self ):
90
99
return parser .Metadata (self .full_name ).build_html ()
Original file line number Diff line number Diff line change 1
1
{% extends "templates/page.jinja" %}
2
2
3
+ {#----------------------------------------------------------------------------#}
4
+
5
+ {% block metadata %}
6
+ {{ builder.top_source_link() }}
7
+
8
+ {{ page_info.doc.brief }}
9
+
10
+ {% if page_info .header %}
11
+
12
+
13
+ {{ page_info.header -}}
14
+ {% endif %}
15
+
16
+ {{ builder.build_collapsable_aliases() -}}
17
+ {% endblock metadata %}
18
+
19
+ {#----------------------------------------------------------------------------#}
20
+
3
21
{% block body %}
4
22
{% if page_info .signature %}
5
23
{{ builder.build_signature() }}
You can’t perform that action at this time.
0 commit comments