rename variable

Author: renatomello

src/qibo/models/_encodings.py

@@ -487,7 +487,7 @@ def _gate_params(
 def _generate_rbs_angles(
     data: ArrayLike,
     architecture: str,
-    nqubits: Optional[int] = None,
+    dims: Optional[int] = None,
     backend: Optional[Backend] = None,
 ) -> List[float]:
     """Generate list of angles for RBS gates based on ``architecture``.
@@ -513,16 +513,16 @@ def _generate_rbs_angles(
         phases.append(backend.arctan2(data[-1], data[-2]))
 
     if architecture == "tree":
-        if nqubits is None:  # pragma: no cover
+        if dims is None:  # pragma: no cover
             raise_error(
                 TypeError,
                 '``nqubits`` must be specified when ``architecture=="tree"``.',
             )
 
-        j_max = int(nqubits / 2)
+        j_max = int(dims / 2)
 
-        r_array = np.zeros(nqubits - 1, dtype=float)
-        phases = np.zeros(nqubits - 1, dtype=float)
+        r_array = np.zeros(dims - 1, dtype=float)
+        phases = np.zeros(dims - 1, dtype=float)
         for j in range(1, j_max + 1):
             r_array[j_max + j - 2] = math.sqrt(
                 data[2 * j - 1] ** 2 + data[2 * j - 2] ** 2

tests/test_quantum_info_metrics.py

@@ -413,8 +413,9 @@ def test_qfim(backend, nqubits, return_complex, params_flag):
         data = np.random.rand(nqubits)
         data = backend.cast(data, dtype=data.dtype)
 
-        params = _generate_rbs_angles(data, nqubits, "diagonal")
-        params = backend.cast(params, dtype=np.float64)
+        params = _generate_rbs_angles(
+            data, dims=nqubits, architecture="diagonal", backend=backend
+        )
 
         target = [1]
         for param in params[:-1]: