Updating stable/0.4 docs and tests due to external dependency updates (…

…#346)
qiskit-community · Mar 18, 2024 · 04773a9 · 04773a9
1 parent 91427d8
commit 04773a9
Show file tree

Hide file tree

Showing 6 changed files with 19 additions and 9 deletions.
diff --git a/docs/tutorials/dynamics_backend.rst b/docs/tutorials/dynamics_backend.rst
@@ -386,10 +386,10 @@ experiments.
     from qiskit_experiments.library.calibration import RoughXSXAmplitudeCal
 
     # rabi experiments for qubit 0
-    rabi0 = RoughXSXAmplitudeCal(0, cals, backend=backend, amplitudes=np.linspace(-0.2, 0.2, 27))
+    rabi0 = RoughXSXAmplitudeCal([0], cals, backend=backend, amplitudes=np.linspace(-0.2, 0.2, 27))
 
     # rabi experiments for qubit 1
-    rabi1 = RoughXSXAmplitudeCal(1, cals, backend=backend, amplitudes=np.linspace(-0.2, 0.2, 27))
+    rabi1 = RoughXSXAmplitudeCal([1], cals, backend=backend, amplitudes=np.linspace(-0.2, 0.2, 27))
 
 Run the Rabi experiments.
 
@@ -425,8 +425,8 @@ as above.
 
     from qiskit_experiments.library.calibration import RoughDragCal
 
-    cal_drag0 = RoughDragCal(0, cals, backend=backend, betas=np.linspace(-20, 20, 15))
-    cal_drag1 = RoughDragCal(1, cals, backend=backend, betas=np.linspace(-20, 20, 15))
+    cal_drag0 = RoughDragCal([0], cals, backend=backend, betas=np.linspace(-20, 20, 15))
+    cal_drag1 = RoughDragCal([1], cals, backend=backend, betas=np.linspace(-20, 20, 15))
 
     cal_drag0.set_experiment_options(reps=[3, 5, 7])
     cal_drag1.set_experiment_options(reps=[3, 5, 7])
@@ -477,8 +477,8 @@ values for the single qubit gates calibrated above.
     from qiskit_experiments.library import CrossResonanceHamiltonian
 
     cr_ham_experiment = CrossResonanceHamiltonian(
-        qubits=(0, 1), 
-        flat_top_widths=np.linspace(0, 5000, 17), 
+        physical_qubits=(0, 1), 
+        durations=np.linspace(1e-7, 1e-6, 17), 
         backend=backend
     )
 

diff --git a/docs/tutorials/optimizing_pulse_sequence.rst b/docs/tutorials/optimizing_pulse_sequence.rst
@@ -312,6 +312,9 @@ entry on :ref:`JAX-compatible pulse schedules <how-to use pulse schedules for ja
             )
         )
 
+        # we need to set disable_validation True to enable jax-jitting.
+        pulse.ScalableSymbolicPulse.disable_validation = True
+
         return pulse.ScalableSymbolicPulse(
                 pulse_type="GaussianSquare",
                 duration=230,

diff --git a/docs/userguide/how_to_use_pulse_schedule_for_jax_jit.rst b/docs/userguide/how_to_use_pulse_schedule_for_jax_jit.rst
@@ -119,6 +119,9 @@ JAX-compiled (or more generally, JAX-transformed).
             _amp * sym.exp(sym.I * _angle) * lifted_gaussian(_t, _center, _duration + 1, _sigma)
         )
 
+        # we need to set disable_validation True to enable jax-jitting.
+        pulse.ScalableSymbolicPulse.disable_validation = True
+
         gaussian_pulse = pulse.ScalableSymbolicPulse(
                 pulse_type="Gaussian",
                 duration=160,

diff --git a/test/dynamics/backend/test_qiskit_experiments.py b/test/dynamics/backend/test_qiskit_experiments.py
@@ -89,7 +89,7 @@ def test_RoughFrequencyCal_calibration(self):
         frequencies = np.linspace(freq_estimate - 1e-2, freq_estimate + 1e-2, 51)
 
         self.simple_backend.set_options(seed_simulator=5243234)
-        spec = RoughFrequencyCal(0, cals, frequencies, backend=self.simple_backend)
+        spec = RoughFrequencyCal([0], cals, frequencies, backend=self.simple_backend)
         spec.set_experiment_options(amp=0.05, sigma=80, duration=320)
 
         spec_data = spec.run().block_for_results()

diff --git a/test/dynamics/pulse/test_pulse_to_signals.py b/test/dynamics/pulse/test_pulse_to_signals.py
@@ -377,8 +377,9 @@ def jit_func_instruction_to_signals(amp):
                 (1, sym.And(_time >= 0, _time <= _duration)), (0, True)
             )
             valid_amp_conditions_expr = sym.Abs(_amp) <= 1.0
-            # we can use only SymbolicPulse when jax-jitting
-            # bacause jax-jitting doesn't correspond to validate_parameters in qiskit.pulse.
+            # we need to set disable_validation True to enable jax-jitting.
+            pulse.SymbolicPulse.disable_validation = True
+
             instance = pulse.SymbolicPulse(
                 pulse_type="Constant",
                 duration=5,
@@ -407,6 +408,7 @@ def jit_func_symbolic_pulse(amp):
                 (1, sym.And(_time >= 0, _time <= _duration)), (0, True)
             )
             valid_amp_conditions_expr = sym.Abs(_amp) <= 1.0
+            pulse.SymbolicPulse.disable_validation = True
             instance = pulse.SymbolicPulse(
                 pulse_type="Constant",
                 duration=5,

diff --git a/test/dynamics/solvers/test_solver_classes.py b/test/dynamics/solvers/test_solver_classes.py
@@ -1304,6 +1304,8 @@ def constant_pulse(amp):
                 (1, sym.And(_time >= 0, _time <= _duration)), (0, True)
             )
             valid_amp_conditions_expr = sym.Abs(_amp) <= 1.0
+            # we need to set disable_validation True to enable jax-jitting.
+            pulse.SymbolicPulse.disable_validation = True
             return pulse.SymbolicPulse(
                 pulse_type="Constant",
                 duration=5,