diff --git a/torchquantum/__init__.py b/torchquantum/__init__.py
index 65ae7892..98376237 100644
--- a/torchquantum/__init__.py
+++ b/torchquantum/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.1.6"
+__version__ = "0.1.7"
 __author__ = "Hanrui Wang, Jiannan Cao, Jessica Ding, Jiai Gu, Song Han, Zirui Li, Zhiding Liang, Pengyu Liu, Mohammadreza Tavasoli"
 
 from .macro import *
diff --git a/torchquantum/__version__.py b/torchquantum/__version__.py
index 6db5a08c..d50bdb0a 100644
--- a/torchquantum/__version__.py
+++ b/torchquantum/__version__.py
@@ -33,4 +33,4 @@
 #         packages=find_packages(),
 #     )
 
-version = "0.1.6"
+version = "0.1.7"
diff --git a/torchquantum/measurement.py b/torchquantum/measurement.py
index b9be5af0..00f0ba4e 100644
--- a/torchquantum/measurement.py
+++ b/torchquantum/measurement.py
@@ -108,6 +108,8 @@ def expval(
 
 
 class MeasureAll(tq.QuantumModule):
+    """Obtain the expectation value of all the qubits.
+    """
     def __init__(self, obs, v_c_reg_mapping=None):
         super().__init__()
         self.obs = obs