Merge branch 'sym' of github.com:qiboteam/qiboml into sym

qiboteam · May 17, 2024 · b97edae · b97edae
2 parents a582cd3 + 26f6d8e
commit b97edae
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/src/qiboml/operations/differentiation.py b/src/qiboml/operations/differentiation.py
@@ -119,22 +119,17 @@ def parameter_shift(
             with respect to circuit's variational parameters.
 
     """
-    nparams = len(circuit.get_parameters())
-    gradient = []
-
-    for i in range(nparams):
-        gradient.append(
-            _one_parameter_shift(
+    return [
+        _one_parameter_shift(
                 hamiltonian=hamiltonian,
                 circuit=circuit,
                 parameter_index=i,
                 initial_state=initial_state,
                 nshots=nshots,
                 exec_backend=exec_backend,
             )
-        )
-
-    return gradient
+        for i in range(len(circuit.get_parameters()))
+    ]
 
 
 def symbolical(