new black

refraction-ray · refraction-ray · commit 86026fa186db · 2024-01-29T20:40:45.000+08:00
diff --git a/docs/source/tutorials/barren_plateaus_cn.ipynb b/docs/source/tutorials/barren_plateaus_cn.ipynb
@@ -158,7 +158,9 @@
     "    dtype=\"float32\",\n",
     ")\n",
     "\n",
-    "e, grad = op_expectation_vmap_vvag(params, seed, n, nlayers)  # 不同随机电路的 ZZ 可观测量和梯度的期望\n",
+    "e, grad = op_expectation_vmap_vvag(\n",
+    "    params, seed, n, nlayers\n",
+    ")  # 不同随机电路的 ZZ 可观测量和梯度的期望\n",
     "\n",
     "grad_var = tf.math.reduce_std(tf.math.reduce_std(grad, axis=0), axis=0)[\n",
     "    0, 0\n",
diff --git a/docs/source/tutorials/nnvqe_cn.ipynb b/docs/source/tutorials/nnvqe_cn.ipynb
@@ -115,7 +115,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "def MERA(inp, n, d=1, lamb=1.0, energy_flag=False):  # 对于单变量一维XXZ模型，我们固定lamb\n",
+    "def MERA(\n",
+    "    inp, n, d=1, lamb=1.0, energy_flag=False\n",
+    "):  # 对于单变量一维XXZ模型，我们固定lamb\n",
     "    params = K.cast(inp[\"params\"], \"complex128\")\n",
     "    delta = K.cast(inp[\"delta\"], \"complex128\")\n",
     "    c = tc.Circuit(n)\n",
diff --git a/docs/source/tutorials/optimization_and_expressibility_cn.ipynb b/docs/source/tutorials/optimization_and_expressibility_cn.ipynb
@@ -276,7 +276,9 @@
     "@tf.function\n",
     "def entropy(v):\n",
     "    ψ = get_state(v)\n",
-    "    ρ_reduced = tc.quantum.reduced_density_matrix(ψ, NA)  # 部分描绘出子系统A得到的降密度矩阵\n",
+    "    ρ_reduced = tc.quantum.reduced_density_matrix(\n",
+    "        ψ, NA\n",
+    "    )  # 部分描绘出子系统A得到的降密度矩阵\n",
     "    S = tc.quantum.renyi_entropy(ρ_reduced)  # Renyi 纠缠熵\n",
     "    return S"
    ]
diff --git a/docs/source/tutorials/tfim_vqe_diffreph_cn.ipynb b/docs/source/tutorials/tfim_vqe_diffreph_cn.ipynb
@@ -420,7 +420,9 @@
     "\n",
     "Jx = np.array([1.0 for _ in range(n - 1)])  # xx 相互作用的强度 (OBC)\n",
     "Bz = np.array([1.0 for _ in range(n)])  # 横向场强\n",
-    "hamiltonian_mpo = tn.matrixproductstates.mpo.FiniteTFI(Jx, Bz, dtype=dtype)  # 矩阵乘积算子\n",
+    "hamiltonian_mpo = tn.matrixproductstates.mpo.FiniteTFI(\n",
+    "    Jx, Bz, dtype=dtype\n",
+    ")  # 矩阵乘积算子\n",
     "hamiltonian_mpo = quoperator_mpo(hamiltonian_mpo)  # 从 mpo 生成 QuOperator"
    ]
   },
diff --git a/examples/matprod_vmap.py b/examples/matprod_vmap.py
@@ -2,6 +2,7 @@
 matrix product: a new twist
 rewrite matrix product in a vmap style
 """
+
 from functools import partial
 
 import numpy as np
diff --git a/tensorcircuit/abstractcircuit.py b/tensorcircuit/abstractcircuit.py
@@ -1,6 +1,7 @@
 """
 Methods for abstract circuits independent of nodes, edges and contractions
 """
+
 # pylint: disable=invalid-name
 
 from typing import Any, Callable, Dict, List, Optional, Sequence, Union, Tuple
diff --git a/tensorcircuit/applications/dqas.py b/tensorcircuit/applications/dqas.py
@@ -1,6 +1,7 @@
 """
 Modules for DQAS framework
 """
+
 # possibly deprecated, multiprocessing is not the recommended way to do DQAS task now, using vmap!
 
 import sys
@@ -486,9 +487,9 @@ def qaoa_simple_train(
     if "prob_model_func" in kws:
         pmf = kws["prob_model_func"]
         del kws["prob_model_func"]
-        kws[
-            "prob_model"
-        ] = pmf()  # in case keras model cannot pickled for multiprocessing map
+        kws["prob_model"] = (
+            pmf()
+        )  # in case keras model cannot pickled for multiprocessing map
     if isinstance(graph, list):
 
         def graph_generator() -> Iterator[Graph]:
diff --git a/tensorcircuit/applications/layers.py b/tensorcircuit/applications/layers.py
@@ -1,6 +1,7 @@
 """
 Module for functions adding layers of circuits
 """
+
 import itertools
 import logging
 import sys
diff --git a/tensorcircuit/applications/physics/fss.py b/tensorcircuit/applications/physics/fss.py
@@ -1,6 +1,7 @@
 """
 finite size scaling tools
 """
+
 from typing import List, Tuple, Optional
 
 import numpy as np
diff --git a/tensorcircuit/applications/utils.py b/tensorcircuit/applications/utils.py
@@ -2,6 +2,7 @@
 A collection of useful function snippets that irrelevant with the main modules
 or await for further refactor
 """
+
 import logging
 from typing import Any, Callable, Iterator, Optional, Sequence, Tuple
 
diff --git a/tensorcircuit/applications/vags.py b/tensorcircuit/applications/vags.py
@@ -1,6 +1,7 @@
 """
 DQAS application kernels as vag functions
 """
+
 from functools import lru_cache, partial, reduce
 import logging
 import operator
diff --git a/tensorcircuit/applications/vqes.py b/tensorcircuit/applications/vqes.py
@@ -1,6 +1,7 @@
 """
 VQNHE application
 """
+
 from functools import lru_cache
 from itertools import product
 import time
diff --git a/tensorcircuit/asciiart.py b/tensorcircuit/asciiart.py
@@ -1,6 +1,7 @@
 """
 Some ascii art from https://www.asciiart.eu/, have fun!
 """
+
 # pylint: disable=invalid-name
 
 import hashlib
diff --git a/tensorcircuit/backends/abstract_backend.py b/tensorcircuit/backends/abstract_backend.py
@@ -1,6 +1,7 @@
 """
 Backend magic inherited from tensornetwork: abstract backend
 """
+
 # pylint: disable=invalid-name
 # pylint: disable=unused-variable
 
@@ -1539,16 +1540,19 @@ def wrapper(*args: Any, **kws: Any) -> Any:
                     args,
                     tuple(
                         [
-                            self.reshape(
-                                self.eye(self.sizen(arg), dtype=arg.dtype),
-                                [-1] + list(self.shape_tuple(arg)),
-                            )
-                            if i == argnum
-                            else self.reshape(
-                                self.zeros(
-                                    [self.sizen(arg), self.sizen(arg)], dtype=arg.dtype
-                                ),
-                                [-1] + list(self.shape_tuple(arg)),
+                            (
+                                self.reshape(
+                                    self.eye(self.sizen(arg), dtype=arg.dtype),
+                                    [-1] + list(self.shape_tuple(arg)),
+                                )
+                                if i == argnum
+                                else self.reshape(
+                                    self.zeros(
+                                        [self.sizen(arg), self.sizen(arg)],
+                                        dtype=arg.dtype,
+                                    ),
+                                    [-1] + list(self.shape_tuple(arg)),
+                                )
                             )
                             for i, arg in enumerate(args)
                         ]
@@ -1607,16 +1611,18 @@ def _first(x: Sequence[Any]) -> Any:
                     args,
                     collect(
                         [
-                            self.reshape(
-                                self.eye(self.sizen(v), dtype=v.dtype),
-                                [-1] + list(self.shape_tuple(v)),
-                            )
-                            if i == k
-                            else self.reshape(
-                                self.zeros(
-                                    [self.sizen(v), self.sizen(v)], dtype=v.dtype
-                                ),
-                                [-1] + list(self.shape_tuple(v)),
+                            (
+                                self.reshape(
+                                    self.eye(self.sizen(v), dtype=v.dtype),
+                                    [-1] + list(self.shape_tuple(v)),
+                                )
+                                if i == k
+                                else self.reshape(
+                                    self.zeros(
+                                        [self.sizen(v), self.sizen(v)], dtype=v.dtype
+                                    ),
+                                    [-1] + list(self.shape_tuple(v)),
+                                )
                             )
                             for i, v in enumerate(values)
                         ]
diff --git a/tensorcircuit/backends/cupy_backend.py b/tensorcircuit/backends/cupy_backend.py
@@ -1,6 +1,7 @@
 """
 CuPy backend. Not in the tensornetwork package and highly experimental.
 """
+
 # pylint: disable=invalid-name
 
 import logging
diff --git a/tensorcircuit/backends/jax_backend.py b/tensorcircuit/backends/jax_backend.py
@@ -1,6 +1,7 @@
 """
 Backend magic inherited from tensornetwork: jax backend
 """
+
 # pylint: disable=invalid-name
 
 from functools import partial
diff --git a/tensorcircuit/backends/jax_ops.py b/tensorcircuit/backends/jax_ops.py
@@ -1,6 +1,7 @@
 """
 Customized ops for ML framework
 """
+
 # pylint: disable=invalid-name
 
 from typing import Any, Tuple, Sequence
diff --git a/tensorcircuit/backends/numpy_backend.py b/tensorcircuit/backends/numpy_backend.py
@@ -1,6 +1,7 @@
 """
 Backend magic inherited from tensornetwork: numpy backend
 """
+
 # pylint: disable=invalid-name
 
 import logging
diff --git a/tensorcircuit/backends/pytorch_backend.py b/tensorcircuit/backends/pytorch_backend.py
@@ -1,6 +1,7 @@
 """
 Backend magic inherited from tensornetwork: pytorch backend
 """
+
 # pylint: disable=invalid-name
 
 import logging
diff --git a/tensorcircuit/backends/pytorch_ops.py b/tensorcircuit/backends/pytorch_ops.py
@@ -1,6 +1,7 @@
 """
 Customized ops for ML framework
 """
+
 # pylint: disable=invalid-name
 
 from typing import Any
diff --git a/tensorcircuit/backends/tensorflow_backend.py b/tensorcircuit/backends/tensorflow_backend.py
@@ -1,6 +1,7 @@
 """
 Backend magic inherited from tensornetwork: tensorflow backend
 """
+
 # pylint: disable=invalid-name
 
 import os
diff --git a/tensorcircuit/backends/tf_ops.py b/tensorcircuit/backends/tf_ops.py
@@ -1,6 +1,7 @@
 """
 Customized ops for ML framework
 """
+
 # pylint: disable=invalid-name
 
 from typing import Any
diff --git a/tensorcircuit/basecircuit.py b/tensorcircuit/basecircuit.py
@@ -1,6 +1,7 @@
 """
 Quantum circuit: common methods for all circuit classes as MixIn
 """
+
 # pylint: disable=invalid-name
 
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Union
diff --git a/tensorcircuit/circuit.py b/tensorcircuit/circuit.py
@@ -1,6 +1,7 @@
 """
 Quantum circuit: the state simulator
 """
+
 # pylint: disable=invalid-name
 
 from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple
diff --git a/tensorcircuit/cloud/utils.py b/tensorcircuit/cloud/utils.py
@@ -1,6 +1,7 @@
 """
 utility functions for cloud connection
 """
+
 from typing import Any, Callable, Optional
 from functools import wraps
 import inspect
diff --git a/tensorcircuit/cloud/wrapper.py b/tensorcircuit/cloud/wrapper.py
@@ -1,6 +1,7 @@
 """
 higher level wrapper shortcut for submit_task
 """
+
 from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 import logging
 import time
diff --git a/tensorcircuit/compiler/__init__.py b/tensorcircuit/compiler/__init__.py
@@ -2,6 +2,7 @@
 Experimental module, no software agnostic unified interface for now,
 only reserve for internal use
 """
+
 from .composed_compiler import Compiler, DefaultCompiler, default_compile
 from . import simple_compiler
 from . import qiskit_compiler
diff --git a/tensorcircuit/cons.py b/tensorcircuit/cons.py
@@ -1,6 +1,7 @@
 """
 Constants and setups
 """
+
 # pylint: disable=invalid-name
 
 import logging
diff --git a/tensorcircuit/densitymatrix.py b/tensorcircuit/densitymatrix.py
@@ -1,6 +1,7 @@
 """
 Quantum circuit class but with density matrix simulator
 """
+
 # pylint: disable=invalid-name
 
 from functools import reduce
@@ -333,9 +334,11 @@ def apply_general_kraus(
     ) -> None:
         # incompatible API for now
         kraus = [
-            k
-            if isinstance(k, tn.Node)
-            else Gate(backend.cast(backend.convert_to_tensor(k), dtypestr))
+            (
+                k
+                if isinstance(k, tn.Node)
+                else Gate(backend.cast(backend.convert_to_tensor(k), dtypestr))
+            )
             for k in kraus
         ]
         self.check_kraus(kraus)
diff --git a/tensorcircuit/fgs.py b/tensorcircuit/fgs.py
@@ -1,6 +1,7 @@
 """
 Fermion Gaussian state simulator
 """
+
 from typing import Any, Dict, List, Optional, Tuple
 
 import numpy as np
diff --git a/tensorcircuit/mps_base.py b/tensorcircuit/mps_base.py
@@ -1,6 +1,7 @@
 """
 FiniteMPS from tensornetwork with bug fixed
 """
+
 # pylint: disable=invalid-name
 
 from typing import Any, Optional, List, Sequence
diff --git a/tensorcircuit/mpscircuit.py b/tensorcircuit/mpscircuit.py
@@ -1,6 +1,7 @@
 """
 Quantum circuit: MPS state simulator
 """
+
 # pylint: disable=invalid-name
 
 from functools import reduce
diff --git a/tensorcircuit/noisemodel.py b/tensorcircuit/noisemodel.py
@@ -1,6 +1,7 @@
 """
 General Noise Model Construction.
 """
+
 import logging
 from functools import partial
 from typing import Any, Sequence, Optional, List, Dict, Tuple, Callable, Union
diff --git a/tensorcircuit/quantum.py b/tensorcircuit/quantum.py
@@ -7,6 +7,7 @@
 
     import tensorcircuit.quantum as qu
 """
+
 # pylint: disable=invalid-name
 
 import os
diff --git a/tensorcircuit/results/counts.py b/tensorcircuit/results/counts.py
@@ -1,6 +1,7 @@
 """
 dict related functionalities
 """
+
 from typing import Any, Dict, Optional, Sequence
 
 import numpy as np
diff --git a/tensorcircuit/results/readout_mitigation.py b/tensorcircuit/results/readout_mitigation.py
diff --git a/tensorcircuit/simplify.py b/tensorcircuit/simplify.py
diff --git a/tensorcircuit/templates/blocks.py b/tensorcircuit/templates/blocks.py
diff --git a/tensorcircuit/templates/graphs.py b/tensorcircuit/templates/graphs.py
diff --git a/tensorcircuit/templates/measurements.py b/tensorcircuit/templates/measurements.py
diff --git a/tensorcircuit/vis.py b/tensorcircuit/vis.py
