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

qiboteam · Jan 3, 2025 · 095e0dd · 095e0dd
2 parents 7c9b764 + 32ec57d
commit 095e0dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_differentiation_rules.py b/tests/test_differentiation_rules.py
@@ -77,7 +77,7 @@ def test_expval_grad_PSR(frontend, backend, nshots):
 
     nparams = len(training_layer.get_parameters())
     initial_params = np.linspace(0.0, 2 * np.pi, nparams)
-    training_layer.set_parameters(initial_params)
+    training_layer.set_parameters(backend.cast(initial_params))
 
     q_model = QuantumModel(
         encoding=encoding_layer,