2424from qiskit .circuit .library import real_amplitudes
2525from qiskit .primitives import StatevectorEstimator
2626
27- from qiskit_algorithms .gradients import DerivativeType , LinCombQGT , ReverseQGT
27+ from qiskit_machine_learning .gradients import DerivativeType , LinCombQGT
2828
2929from .logging_primitives import LoggingEstimator
3030
@@ -37,10 +37,10 @@ def setUp(self):
3737 super ().setUp ()
3838 self .estimator = StatevectorEstimator ()
3939
40- @data (LinCombQGT , ReverseQGT )
40+ @data (LinCombQGT )
4141 def test_qgt_derivative_type (self , qgt_type ):
4242 """Test QGT derivative_type"""
43- args = () if qgt_type == ReverseQGT else ( self . estimator ,)
43+ args = ()
4444 qgt = qgt_type (* args , derivative_type = DerivativeType .REAL )
4545
4646 a , b = Parameter ("a" ), Parameter ("b" )
@@ -76,10 +76,10 @@ def test_qgt_derivative_type(self, qgt_type):
7676 qgt_result = qgt .run ([qc ], [param ]).result ().qgts
7777 np .testing .assert_allclose (qgt_result [0 ], correct_values [i ], atol = 1e-3 )
7878
79- @data (LinCombQGT , ReverseQGT )
79+ @data (LinCombQGT )
8080 def test_qgt_phase_fix (self , qgt_type ):
8181 """Test the phase-fix argument in a QGT calculation"""
82- args = () if qgt_type == ReverseQGT else ( self . estimator ,)
82+ args = ()
8383 qgt = qgt_type (* args , phase_fix = False )
8484
8585 # create the circuit
@@ -116,10 +116,10 @@ def test_qgt_phase_fix(self, qgt_type):
116116 qgt_result = qgt .run ([qc ], [param ]).result ().qgts
117117 np .testing .assert_allclose (qgt_result [0 ], correct_values [i ], atol = 1e-3 )
118118
119- @data (LinCombQGT , ReverseQGT )
119+ @data (LinCombQGT )
120120 def test_qgt_coefficients (self , qgt_type ):
121121 """Test the derivative option of QGT"""
122- args = () if qgt_type == ReverseQGT else ( self . estimator ,)
122+ args = ()
123123 qgt = qgt_type (* args , derivative_type = DerivativeType .REAL )
124124
125125 qc = real_amplitudes (num_qubits = 2 , reps = 1 )
@@ -154,10 +154,10 @@ def test_qgt_coefficients(self, qgt_type):
154154 qgt_result = qgt .run ([qc ], [param ]).result ().qgts
155155 np .testing .assert_allclose (qgt_result [0 ], correct_values [i ], atol = 1e-3 )
156156
157- @data (LinCombQGT , ReverseQGT )
157+ @data (LinCombQGT )
158158 def test_qgt_parameters (self , qgt_type ):
159159 """Test the QGT with specified parameters"""
160- args = () if qgt_type == ReverseQGT else ( self . estimator ,)
160+ args = ()
161161 qgt = qgt_type (* args , derivative_type = DerivativeType .REAL )
162162
163163 a = Parameter ("a" )
@@ -199,10 +199,10 @@ def test_qgt_parameters(self, qgt_type):
199199 qgt_result = qgt .run ([qc2 ], [param_values ], [param ]).result ().qgts
200200 np .testing .assert_allclose (qgt_result [0 ], expected [i ], atol = 1e-3 )
201201
202- @data (LinCombQGT , ReverseQGT )
202+ @data (LinCombQGT )
203203 def test_qgt_multi_arguments (self , qgt_type ):
204204 """Test the QGT for multiple arguments"""
205- args = () if qgt_type == ReverseQGT else ( self . estimator ,)
205+ args = ()
206206 qgt = qgt_type (* args , derivative_type = DerivativeType .REAL )
207207
208208 a = Parameter ("a" )
@@ -221,10 +221,10 @@ def test_qgt_multi_arguments(self, qgt_type):
221221 for i , _ in enumerate (param_list ):
222222 np .testing .assert_allclose (qgt_results [i ], correct_values [i ], atol = 1e-3 )
223223
224- @data (LinCombQGT , ReverseQGT )
224+ @data (LinCombQGT )
225225 def test_qgt_validation (self , qgt_type ):
226226 """Test estimator QGT's validation"""
227- args = () if qgt_type == ReverseQGT else ( self . estimator ,)
227+ args = ()
228228 qgt = qgt_type (* args )
229229
230230 a = Parameter ("a" )
0 commit comments