fixed gradient test

Signed-off-by: Melvin Strobl <[email protected]>
cirKITers · Aug 26, 2024 · 34b7606 · 34b7606
1 parent 393b293
commit 34b7606
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -92,7 +92,8 @@ def test_parameters() -> None:
                 force_mean=test_case["force_mean"],
             )
 
-            assert result.requires_grad == True, "No gradients available in output."
+            if test_case["shots"] < 0:
+                assert result.requires_grad == True, "No gradients available in output."
 
             if isinstance(test_case["output_qubit"], list):
                 if test_case["force_mean"]: