From 122e7be07b93ac4dce081711dc7a8ed0ce150eb2 Mon Sep 17 00:00:00 2001 From: liuchao Date: Fri, 17 Oct 2025 16:25:22 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dmint.optim.Adam=E6=A0=B7?= =?UTF-8?q?=E4=BE=8B=E4=B8=AD=E9=94=99=E6=8A=8Ann=E6=94=B9=E6=88=90?= =?UTF-8?q?=E4=BA=86mint=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/mindspore/_ext/overwrite_autodoc.txt | 5 ++++- docs/mindspore/_ext/overwriteobjectiondirective.txt | 3 +++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/docs/mindspore/_ext/overwrite_autodoc.txt b/docs/mindspore/_ext/overwrite_autodoc.txt index f79d257f4c..2a8d580b30 100644 --- a/docs/mindspore/_ext/overwrite_autodoc.txt +++ b/docs/mindspore/_ext/overwrite_autodoc.txt @@ -1796,7 +1796,10 @@ class ClassDocumenter(DocstringSignatureMixin, ModuleLevelDocumenter): # type: mint_doc = re.sub(r'(\n[ ]+(-|\*) GPU(/CPU)?:.*\n|\n[ ]+(-|\*) CPU(/GPU)?:.*\n)', '\n', docstrings[i]) mint_doc = re.sub(r'(Alias for.*?:(func|class):`[^`]+?\.)ops\.([^`]+?)(?>> from mindspore import .*?)(ops|nn)', r'\1mint', mint_doc) + if 'mindspore/mint/optim' in py_source_rel: + mint_doc = re.sub('(>>> from mindspore import .*?)(ops)', r'\1mint', mint_doc) + else: + mint_doc = re.sub('(>>> from mindspore import .*?)(ops|nn)', r'\1mint', mint_doc) mint_doc = mint_doc.replace('>>> import mindspore.ops as ops', '>>> from mindspore import mint') if not re.findall(f'>>> .*?{usename}', mint_doc): mint_doc = re.sub(rf'(>>> .*?)(ops|nn)\.[\w\.]*?{b_name}(Ext)?', rf'\1{usename}', mint_doc) diff --git a/docs/mindspore/_ext/overwriteobjectiondirective.txt b/docs/mindspore/_ext/overwriteobjectiondirective.txt index e2ab339e62..21795b6879 100644 --- a/docs/mindspore/_ext/overwriteobjectiondirective.txt +++ b/docs/mindspore/_ext/overwriteobjectiondirective.txt @@ -90,6 +90,9 @@ def get_example(name: str): if re.findall(rf'>>> .*?ops\.\w+\(', api_doc): api_doc = re.sub('(>>> from mindspore import .*?)(ops|nn)', r'\1mint, ops', api_doc) api_doc = api_doc.replace('>>> import mindspore.ops as ops', '>>> from mindspore import mint, ops') + elif 'mint.optim' in usename: + api_doc = re.sub('(>>> from mindspore import .*?)(ops)', r'\1mint', api_doc) + api_doc = api_doc.replace('>>> import mindspore.ops as ops', '>>> from mindspore import mint') else: api_doc = re.sub('(>>> from mindspore import .*?)(ops|nn)', r'\1mint', api_doc) api_doc = api_doc.replace('>>> import mindspore.ops as ops', '>>> from mindspore import mint') -- Gitee