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,
return_annotation):
if signature:
signature = re.sub('<Mock name=\'([^\']+)\'.*>', '\g<1>', signature)
signature = re.sub('tensorflow', 'tf')
signature = re.sub('tensorflow', 'tf', signature)
return signature, return_annotation
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