From 04cf66848b205c45aec72c00dcd305d9d2a788ce Mon Sep 17 00:00:00 2001
From: lvmingfu <lvmingfu@huawei.com>
Date: Thu, 29 Sep 2022 11:34:53 +0800
Subject: [PATCH] modify context code formats

---
 mindarmour/privacy/sup_privacy/mask_monitor/masker.py | 10 +++++-----
 mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py    | 10 +++++-----
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/mindarmour/privacy/sup_privacy/mask_monitor/masker.py b/mindarmour/privacy/sup_privacy/mask_monitor/masker.py
index 18cb07d..6eef721 100644
--- a/mindarmour/privacy/sup_privacy/mask_monitor/masker.py
+++ b/mindarmour/privacy/sup_privacy/mask_monitor/masker.py
@@ -36,8 +36,8 @@ class SuppressMasker(Callback):
 
     Examples:
         >>> import mindspore.nn as nn
-        >>> import mindspore.ops.operations as P
-        >>> from mindspore import context
+        >>> import mindspore as ms
+        >>> from mindspore import set_context
         >>> from mindspore.nn import Accuracy
         >>> from mindarmour.privacy.sup_privacy import SuppressModel
         >>> from mindarmour.privacy.sup_privacy import SuppressMasker
@@ -46,14 +46,14 @@ class SuppressMasker(Callback):
         >>> class Net(nn.Cell):
         ...     def __init__(self):
         ...         super(Net, self).__init__()
-        ...         self._softmax = P.Softmax()
+        ...         self._softmax = ops.Softmax()
         ...         self._Dense = nn.Dense(10,10)
-        ...         self._squeeze = P.Squeeze(1)
+        ...         self._squeeze = ops.Squeeze(1)
         ...     def construct(self, inputs):
         ...         out = self._softmax(inputs)
         ...         out = self._Dense(out)
         ...         return self._squeeze(out)
-        >>> context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
+        >>> set_context(mode=ms.PYNATIVE_MODE, device_target="GPU")
         >>> network = Net()
         >>> masklayers = []
         >>> masklayers.append(MaskLayerDes("_Dense.weight", 0, False, True, 10))
diff --git a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py
index f3e0e6c..e6209c8 100644
--- a/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py
+++ b/mindarmour/privacy/sup_privacy/sup_ctrl/conctrl.py
@@ -66,8 +66,8 @@ class SuppressPrivacyFactory:
 
         Examples:
             >>> import mindspore.nn as nn
-            >>> import mindspore.ops.operations as P
-            >>> from mindspore import context
+            >>> import mindspore as ms
+            >>> from mindspore import set_context, ops
             >>> from mindspore.nn import Accuracy
             >>> from mindarmour.privacy.sup_privacy import SuppressPrivacyFactory
             >>> from mindarmour.privacy.sup_privacy import MaskLayerDes
@@ -75,14 +75,14 @@ class SuppressPrivacyFactory:
             >>> class Net(nn.Cell):
             ...     def __init__(self):
             ...         super(Net, self).__init__()
-            ...         self._softmax = P.Softmax()
+            ...         self._softmax = ops.Softmax()
             ...         self._Dense = nn.Dense(10,10)
-            ...         self._squeeze = P.Squeeze(1)
+            ...         self._squeeze = ops.Squeeze(1)
             ...     def construct(self, inputs):
             ...         out = self._softmax(inputs)
             ...         out = self._Dense(out)
             ...         return self._squeeze(out)
-            >>> context.set_context(mode=context.PYNATIVE_MODE, device_target="CPU")
+            >>> set_context(mode=ms.PYNATIVE_MODE, device_target="CPU")
             >>> network = Net()
             >>> masklayers = []
             >>> masklayers.append(MaskLayerDes("_Dense.weight", 0, False, True, 10))