fixing validation disablement

qiskit-community · Mar 18, 2024 · 5f18adf · 5f18adf
1 parent 7272b4f
commit 5f18adf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/test/dynamics/pulse/test_pulse_to_signals.py b/test/dynamics/pulse/test_pulse_to_signals.py
@@ -378,7 +378,7 @@ def jit_func_instruction_to_signals(amp):
             )
             valid_amp_conditions_expr = sym.Abs(_amp) <= 1.0
             # we need to set disable_validation True to enable jax-jitting.
-            pulse.ScalableSymbolicPulse.disable_validation = True
+            pulse.SymbolicPulse.disable_validation = True
 
             instance = pulse.SymbolicPulse(
                 pulse_type="Constant",
@@ -408,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
@@ -1305,7 +1305,7 @@ def constant_pulse(amp):
             )
             valid_amp_conditions_expr = sym.Abs(_amp) <= 1.0
             # we need to set disable_validation True to enable jax-jitting.
-            pulse.ScalableSymbolicPulse.disable_validation = True
+            pulse.SymbolicPulse.disable_validation = True
             return pulse.SymbolicPulse(
                 pulse_type="Constant",
                 duration=5,