diff --git a/docs/articles_en/assets/snippets/ov_extensions.py b/docs/articles_en/assets/snippets/ov_extensions.py index 621a0da21c5839..8064a5b7163e7f 100644 --- a/docs/articles_en/assets/snippets/ov_extensions.py +++ b/docs/articles_en/assets/snippets/ov_extensions.py @@ -47,7 +47,7 @@ def conversion(node): #! [py_frontend_extension_aten_hardtanh] import torch from openvino.frontend import ConversionExtension, NodeContext -from openvino.tools.mo import convert_model +from openvino import convert_model class HardTanh(torch.nn.Module): @@ -69,5 +69,5 @@ def convert_hardtanh(node: NodeContext): model = HardTanh(min_val=0.1, max_val=2.0) hardtanh_ext = ConversionExtension("aten::hardtanh", convert_hardtanh) -ov_model = convert_model(input_model=model, extensions=[hardtanh_ext]) +ov_model = convert_model(input_model=model, extension=[hardtanh_ext]) #! [py_frontend_extension_aten_hardtanh]