tequila-basic 1.9.8__py3-none-any.whl → 1.9.10__py3-none-any.whl

tequila/__init__.py

@@ -1,25 +1,37 @@
 from tequila.utils import BitString, BitNumbering, BitStringLSB, initialize_bitstring, TequilaException, TequilaWarning
 from tequila.circuit import gates, QCircuit, NoiseModel, compile_circuit, CircuitCompiler
 from tequila.hamiltonian import paulis, QubitHamiltonian, PauliString
-from tequila.objective import Objective, VectorObjective,\
-    ExpectationValue, Variable, assign_variable, format_variable_dictionary,\
-    vectorize
+from tequila.objective import (
+    Objective,
+    VectorObjective,
+    ExpectationValue,
+    Variable,
+    assign_variable,
+    format_variable_dictionary,
+    vectorize,
+)
 from tequila.objective import QTensor
 from tequila.objective.braket import BraKet, make_transition, make_overlap, Overlap, Fidelity
 
-# backward compatibility
-braket = BraKet
-
 from tequila.optimizers import INSTALLED_OPTIMIZERS, show_available_optimizers
 from tequila.optimizers import minimize, minimize_scipy, minimize_gd, optimizer_scipy
 
-from tequila.simulators.simulator_api import simulate, compile, compile_to_function, draw, pick_backend, \
-    INSTALLED_SAMPLERS, \
-    INSTALLED_SIMULATORS, SUPPORTED_BACKENDS, INSTALLED_BACKENDS, show_available_simulators
+from tequila.simulators.simulator_api import (
+    simulate,
+    compile,
+    compile_to_function,
+    draw,
+    pick_backend,
+    INSTALLED_SAMPLERS,
+    INSTALLED_SIMULATORS,
+    SUPPORTED_BACKENDS,
+    INSTALLED_BACKENDS,
+    show_available_simulators,
+)
 from tequila.wavefunction import QubitWaveFunction
 from tequila.circuit.qasm import export_open_qasm, import_open_qasm, import_open_qasm_from_file
 from tequila.circuit.pyzx import convert_to_pyzx, convert_from_pyzx
-import tequila.quantumchemistry as chemistry # shortcut
+import tequila.quantumchemistry as chemistry  # shortcut
 from tequila.quantumchemistry import Molecule, MoleculeFromOpenFermion, MoleculeFromTequila
 
 # make sure to use the jax/autograd numpy for objectives
@@ -32,11 +44,14 @@ from tequila.tools.random_generators import make_random_circuit, make_random_ham
 # get rid of the jax GPU/CPU warnings
 import warnings
 
-warnings.filterwarnings("ignore", module="jax")
-warnings.filterwarnings("ignore", module="absl")
-warnings.filterwarnings("default", category=TequilaWarning)
-
 # load applications and helpers
 from tequila.apps import adapt
 
 from .version import __version__, __author__
+
+# backward compatibility
+braket = BraKet
+
+warnings.filterwarnings("ignore", module="jax")
+warnings.filterwarnings("ignore", module="absl")
+warnings.filterwarnings("default", category=TequilaWarning)

tequila/apps/__init__.py

@@ -3,7 +3,8 @@ from tequila.apps.unary_state_prep import UnaryStatePrep
 from tequila.apps import adapt
 from tequila.apps import krylov
 
-def gem(circuits:list, H, variables:dict=None, *args, **kwargs):
+
+def gem(circuits: list, H, variables: dict = None, *args, **kwargs):
     """
     Solve the generalized eigenvalue equation: HC = ESC
     with matries Hij = <i|H|j>, Sij = <i|j> and states |i> = U[i]|00...0> generated by the circuits given
@@ -24,13 +25,21 @@ def gem(circuits:list, H, variables:dict=None, *args, **kwargs):
     ----------
     a tuple of eigenenergies E and eigenvectors C in the same format as numpy.linalg.eigh
     """
-    
+
     varnames = sum([U.extract_variables() for U in circuits], [])
     if len(varnames) > 0:
         if variables is None:
-            raise TequilaException("GEM method: no variables dictionary given, but circuits have {} variables:\n{}".format(len(varnames),str(varnames)))
+            raise TequilaException(
+                "GEM method: no variables dictionary given, but circuits have {} variables:\n{}".format(
+                    len(varnames), str(varnames)
+                )
+            )
         for k in varnames:
             if k not in variables:
-                raise TequilaException("GEM method: you gave a parametrized circuit, but the variable {} was not set in the variables dictionary".format(k))
+                raise TequilaException(
+                    "GEM method: you gave a parametrized circuit, but the variable {} was not set in the variables dictionary".format(
+                        k
+                    )
+                )
 
-    return krylov.krylov_method(krylov_circs=circuits,H=H,variables=variables) 
+    return krylov.krylov_method(krylov_circs=circuits, H=H, variables=variables)