Commit b5495059 authored by Yuxin Wu's avatar Yuxin Wu

naive bug

parent 95ec24d3
...@@ -317,7 +317,7 @@ def get_rst(app, what, name, obj, options, signature, ...@@ -317,7 +317,7 @@ def get_rst(app, what, name, obj, options, signature,
return_annotation): return_annotation):
if signature: if signature:
signature = re.sub('<Mock name=\'([^\']+)\'.*>', '\g<1>', signature) signature = re.sub('<Mock name=\'([^\']+)\'.*>', '\g<1>', signature)
signature = re.sub('tensorflow', 'tf') signature = re.sub('tensorflow', 'tf', signature)
return signature, return_annotation return signature, return_annotation
def setup(app): def setup(app):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment