fix bug with jax

PennyLaneAI · Jun 5, 2024 · 8d7594a · 8d7594a
1 parent 7b87b2e
commit 8d7594a
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pennylane/workflow/execution.py b/pennylane/workflow/execution.py
@@ -298,16 +298,14 @@ def inner_execute(tapes: Sequence[QuantumTape], **_) -> ResultBatch:
             cache (None | MutableMapping): The cache to use. If ``None``, caching will not occur.
         """
 
-        transform_program = qml.transforms.core.TransformProgram()
+        transform_program = qml.transforms.core.TransformProgram(inner_transform)
 
         if numpy_only:
             transform_program.add_transform(qml.transforms.convert_to_numpy_parameters)
 
         if cache is not None:
             transform_program.add_transform(_cache_transform, cache=cache)
 
-        transform_program += inner_transform
-
         transformed_tapes, transform_post_processing = transform_program(tapes)
 
         # TODO: Apply expand_fn() as transform.